Searched refs:tensor_out_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_avgpooling_op.cc | 124 TensorShape tensor_out_shape; in Compute() local
133 tensor_out_shape.AddDim(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>( in Compute()
137 AllocateOutputSetMklShape(context, 0, &output, tensor_out_shape, in Compute()
314 TensorShape tensor_out_shape; in Compute() local
324 tensor_out_shape.AddDim(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>( in Compute()
328 AllocateOutputSetMklShape(context, 0, &output, tensor_out_shape, in Compute()
|
| D | pooling_ops_common.cc | 159 const TensorShape& tensor_out_shape, in Compute() argument
163 context->allocate_output(0, tensor_out_shape, &tensor_out)); in Compute()
196 ShapeFromFormat(FORMAT_NCHW, tensor_out_shape, in Compute()
|
| D | pooling_ops_common_gpu.h | 47 const TensorShape& tensor_out_shape, bool propagate_nans);
|
| D | mkl_maxpooling_op.cc | 105 TensorShape tensor_out_shape; in Compute() local
115 tensor_out_shape.AddDim(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>( in Compute()
118 AllocateOutputSetMklShape(context, 0, &output_tensor, tensor_out_shape, in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 309 const TensorShape tensor_out_shape = ctx->InputShape(1); in Compile() local
316 OP_REQUIRES(ctx, tensor_out_shape.dims() == num_dims(), in Compile()
516 const TensorShape tensor_out_shape = ctx->InputShape(1); in Compile() local
523 OP_REQUIRES(ctx, tensor_out_shape.dims() == num_dims(), in Compile()
|