| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 277 const Tensor& tensor_in_shape = context->input(0); in Compute() local
282 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute()
294 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute()
295 for (int64 i = 0; i < tensor_in_shape.NumElements(); ++i) { in Compute()
439 const Tensor& tensor_in_shape = context->input(0); in Compute() local
444 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute()
452 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute()
453 for (int64 i = 0; i < tensor_in_shape.NumElements(); ++i) { in Compute()
526 const Tensor& tensor_in_shape = context->input(0); in Compute() local
531 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 std::array<int64, 3> input_size = {{tensor_in_shape.dim_size(3), in launch()
388 tensor_in_shape.dim_size(2), in launch()
[all …]
|
| D | pooling_ops_common.cc | 122 const TensorShape& tensor_in_shape) { in PoolParameters() argument
127 GetTensorSpatialDims(tensor_in_shape.dims(), data_format) == 2, in PoolParameters()
130 tensor_in_shape.dims(), " ", data_format)); in PoolParameters()
133 depth = GetTensorDim(tensor_in_shape, data_format, 'C') * in PoolParameters()
135 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in PoolParameters()
136 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in PoolParameters()
137 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in PoolParameters()
446 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute() argument
455 context->allocate_output(0, tensor_in_shape, &input_backprop)); in Compute()
456 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 in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Compute()
146 const int64 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 | pooling_ops_3d.h | 45 const TensorShape& tensor_in_shape);
|
| D | cudnn_pooling_gpu.h | 61 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 | 315 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local
320 OP_REQUIRES(ctx, tensor_in_shape.dims() == num_dims(), in Compile()
344 XlaTensorFormat(data_format_, tensor_in_shape.dims() - 2)); in Compile()
535 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local
540 OP_REQUIRES(ctx, tensor_in_shape.dims() == num_dims(), in Compile()
|