| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 282 const Tensor& tensor_in_shape = context->input(0); in Compute() local
287 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute()
299 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute()
300 for (int64_t i = 0; i < tensor_in_shape.NumElements(); ++i) { in Compute()
445 const Tensor& tensor_in_shape = context->input(0); in Compute() local
450 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute()
458 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute()
459 for (int64_t i = 0; i < tensor_in_shape.NumElements(); ++i) { in Compute()
532 const Tensor& tensor_in_shape = context->input(0); in Compute() local
537 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute()
[all …]
|
| D | pooling_ops_3d.cc | 52 const TensorShape& tensor_in_shape) { in Pool3dParameters() argument
54 OP_REQUIRES(context, tensor_in_shape.dims() == 5, in Pool3dParameters()
58 depth = GetTensorDim(tensor_in_shape, data_format, 'C'); in Pool3dParameters()
59 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Pool3dParameters()
60 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Pool3dParameters()
61 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); in Pool3dParameters()
62 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Pool3dParameters()
379 const TensorShape& tensor_in_shape, in launch()
387 context, tensor_in_shape.dim_size(0) == out_backprop.dim_size(0), in launch()
391 tensor_in_shape.dim_size(0), " and ", out_backprop.dim_size(0))); in launch()
[all …]
|
| D | pooling_ops_common.cc | 123 const TensorShape& tensor_in_shape) { in PoolParameters() argument
128 GetTensorSpatialDims(tensor_in_shape.dims(), data_format) == 2, in PoolParameters()
131 tensor_in_shape.dims(), " ", data_format)); in PoolParameters()
134 depth = GetTensorDim(tensor_in_shape, data_format, 'C') * in PoolParameters()
136 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in PoolParameters()
137 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in PoolParameters()
138 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in PoolParameters()
449 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute() argument
458 context->allocate_output(0, tensor_in_shape, &input_backprop)); in Compute()
459 if (tensor_in_shape.num_elements() == 0) { in Compute()
[all …]
|
| D | cudnn_pooling_gpu.cc | 138 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute() argument
145 const int64_t in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Compute()
146 const int64_t in_features = GetTensorDim(tensor_in_shape, data_format, 'C'); in Compute()
152 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute()
223 dim_i, GetTensorDim(tensor_in_shape, data_format, '2' - i)); in Compute()
|
| D | cudnn_pooling_gpu.h | 61 const TensorShape& tensor_in_shape,
|
| D | pooling_ops_3d.h | 45 const TensorShape& tensor_in_shape);
|
| D | pooling_ops_common_gpu.h | 65 const TensorShape& tensor_in_shape, bool propagate_nans);
|
| D | pooling_ops_common.h | 53 const TensorShape& tensor_in_shape);
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.cc | 229 const TensorShape& tensor_in_shape) { in Init() argument
232 tensor_in_shape.dims() == 4 || tensor_in_shape.dims() == 5, in Init()
235 depth = GetTensorDim(tensor_in_shape, data_format, 'C'); in Init()
236 if (tensor_in_shape.dims() == 4) { in Init()
238 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in Init()
239 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in Init()
242 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Init()
243 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Init()
244 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); in Init()
246 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Init()
|
| D | mkl_pooling_ops_common.h | 419 TensorFormat data_format, const TensorShape& tensor_in_shape);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 354 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local
359 OP_REQUIRES(ctx, tensor_in_shape.dims() == num_dims(), in Compile()
383 XlaTensorFormat(data_format_, tensor_in_shape.dims() - 2)); in Compile()
576 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local
581 OP_REQUIRES(ctx, tensor_in_shape.dims() == num_dims(), in Compile()
|