Searched refs:stride_shape (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpooling_ops.cc97 const TensorShape stride_shape = ctx->InputShape(2); variable
99 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(stride_shape),
101 stride_shape.DebugString()));
102 OP_REQUIRES(ctx, stride_shape.num_elements() == num_dims(),
300 const TensorShape stride_shape = ctx->InputShape(4); variable
302 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(stride_shape),
304 stride_shape.DebugString()));

Completed in 161 milliseconds