/external/tensorflow/tensorflow/core/kernels/ |
D | pooling_ops_common.cc | 50 const TensorShape& tensor_in_shape) { in PoolParameters() argument 55 GetTensorSpatialDims(tensor_in_shape.dims(), data_format) == 2, in PoolParameters() 58 tensor_in_shape.dims(), " ", data_format)); in PoolParameters() 61 depth = GetTensorDim(tensor_in_shape, data_format, 'C') * in PoolParameters() 63 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in PoolParameters() 64 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in PoolParameters() 65 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in PoolParameters() 290 const TensorShape& tensor_in_shape, bool propagate_nans) { in Compute() argument 298 context->allocate_output(0, tensor_in_shape, &input_backprop)); in Compute() 299 if (tensor_in_shape.num_elements() == 0) { in Compute() [all …]
|
D | avgpooling_op.cc | 243 const Tensor& tensor_in_shape = context->input(0); in Compute() local 248 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute() 260 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() 261 for (int64 i = 0; i < tensor_in_shape.NumElements(); ++i) { in Compute() 402 const Tensor& tensor_in_shape = context->input(0); in Compute() local 407 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute() 415 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() 416 for (int64 i = 0; i < tensor_in_shape.NumElements(); ++i) { in Compute() 482 const Tensor& tensor_in_shape = context->input(0); in Compute() local 487 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute() [all …]
|
D | mkl_pooling_ops_common.cc | 228 const TensorShape& tensor_in_shape) { in Init() argument 231 tensor_in_shape.dims() == 4 || tensor_in_shape.dims() == 5, in Init() 234 depth = GetTensorDim(tensor_in_shape, data_format, 'C'); in Init() 235 if (tensor_in_shape.dims() == 4) { in Init() 237 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in Init() 238 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in Init() 241 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Init() 242 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Init() 243 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); in Init() 245 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Init()
|
D | pooling_ops_3d.cc | 57 const TensorShape& tensor_in_shape) { in Pool3dParameters() argument 59 OP_REQUIRES(context, tensor_in_shape.dims() == 5, in Pool3dParameters() 63 depth = GetTensorDim(tensor_in_shape, data_format, 'C'); in Pool3dParameters() 64 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Pool3dParameters() 65 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Pool3dParameters() 66 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); in Pool3dParameters() 67 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Pool3dParameters() 383 const TensorShape& tensor_in_shape, in launch() 391 std::array<int64, 3> input_size = {{tensor_in_shape.dim_size(3), in launch() 392 tensor_in_shape.dim_size(2), in launch() [all …]
|
D | cudnn_pooling_gpu.cc | 123 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute() argument 130 const int64 in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Compute() 131 const int64 in_features = GetTensorDim(tensor_in_shape, data_format, 'C'); in Compute() 137 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute() 208 dim_i, GetTensorDim(tensor_in_shape, data_format, '2' - i)); in Compute()
|
D | mkl_avgpooling_op.cc | 245 const Tensor& tensor_in_shape = MklGetInput(context, 0); in Compute() local 252 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() 253 for (int64 i = 0; i < tensor_in_shape.NumElements(); ++i) { in Compute() 353 const Tensor& tensor_in_shape = MklGetInput(context, 0); in MklCreateLayoutsAndPrimitives() local 362 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in MklCreateLayoutsAndPrimitives() 673 const Tensor& tensor_in_shape, in SanityCheckInputs() argument 680 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in SanityCheckInputs()
|
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 | 63 const TensorShape& tensor_in_shape, bool propagate_nans);
|
D | pooling_ops_3d_sycl.h | 719 const TensorShape& tensor_in_shape, 727 const int batch = GetTensorDim(tensor_in_shape, data_format, 'N'); 728 const int in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); 729 const int in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); 730 const int in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); 731 const int depth = GetTensorDim(tensor_in_shape, data_format, 'C');
|
D | maxpooling_op.cc | 326 const Tensor& out_backprop, const TensorShape& tensor_in_shape) { in MaxPoolingBackwardCustomKernel() argument 329 {0}, 0, tensor_in_shape, &output)); in MaxPoolingBackwardCustomKernel() 332 padding, FORMAT_NHWC, tensor_in_shape}; in MaxPoolingBackwardCustomKernel()
|
D | pooling_ops_common.h | 45 TensorFormat data_format, const TensorShape& tensor_in_shape);
|
D | mkl_pooling_ops_common.h | 426 TensorFormat data_format, const TensorShape& tensor_in_shape);
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | pooling_ops.cc | 308 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local 313 OP_REQUIRES(ctx, tensor_in_shape.dims() == num_dims(), in Compile() 515 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local 520 OP_REQUIRES(ctx, tensor_in_shape.dims() == num_dims(), in Compile()
|