Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc103 const TensorShape stride_shape = ctx->InputShape(2); in GetStride() local
105 if (!TensorShapeUtils::IsVector(stride_shape)) { in GetStride()
107 stride_shape.DebugString()); in GetStride()
109 if (stride_shape.num_elements() != num_dims()) { in GetStride()
291 const TensorShape stride_shape = ctx->InputShape(4); in Compile() local
293 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(stride_shape), in Compile()
295 stride_shape.DebugString())); in Compile()
498 const TensorShape stride_shape = ctx->InputShape(4); in Compile() local
500 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(stride_shape), in Compile()
502 stride_shape.DebugString())); in Compile()