| /external/tensorflow/tensorflow/core/kernels/ |
| pooling_ops_common_gpu.h | 47 const TensorShape& tensor_out_shape, bool propagate_nans);
|
| pooling_ops_common.cc | 159 const TensorShape& tensor_out_shape,
163 context->allocate_output(0, tensor_out_shape, &tensor_out));
196 ShapeFromFormat(FORMAT_NCHW, tensor_out_shape,
|
| mkl_avgpooling_op.cc | 124 TensorShape tensor_out_shape; variable
133 tensor_out_shape.AddDim(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
137 AllocateOutputSetMklShape(context, 0, &output, tensor_out_shape,
314 TensorShape tensor_out_shape; variable
324 tensor_out_shape.AddDim(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
328 AllocateOutputSetMklShape(context, 0, &output, tensor_out_shape,
|
| mkl_maxpooling_op.cc | 105 TensorShape tensor_out_shape; variable
115 tensor_out_shape.AddDim(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
118 AllocateOutputSetMklShape(context, 0, &output_tensor, tensor_out_shape,
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| pooling_ops.cc | 309 const TensorShape tensor_out_shape = ctx->InputShape(1); variable
316 OP_REQUIRES(ctx, tensor_out_shape.dims() == num_dims(),
516 const TensorShape tensor_out_shape = ctx->InputShape(1); variable
523 OP_REQUIRES(ctx, tensor_out_shape.dims() == num_dims(),
|
