Searched refs:tensor_out_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common_gpu.h | 48 const TensorShape& tensor_out_shape, bool propagate_nans);
|
| D | pooling_ops_common.cc | 219 const TensorShape& tensor_out_shape, in Compute() argument
223 context->allocate_output(0, tensor_out_shape, &tensor_out)); in Compute()
260 ShapeFromFormat(FORMAT_NCHW, tensor_out_shape, in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 316 const TensorShape tensor_out_shape = ctx->InputShape(1); in Compile() local
323 OP_REQUIRES(ctx, tensor_out_shape.dims() == num_dims(), in Compile()
536 const TensorShape tensor_out_shape = ctx->InputShape(1); in Compile() local
543 OP_REQUIRES(ctx, tensor_out_shape.dims() == num_dims(), in Compile()
|