| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 499 data_.map_[GetTensorDimIndex<3>(data_format, '0')] = in SetTfDimOrder()
501 data_.map_[GetTensorDimIndex<3>(data_format, '1')] = in SetTfDimOrder()
503 data_.map_[GetTensorDimIndex<3>(data_format, '2')] = in SetTfDimOrder()
505 data_.map_[GetTensorDimIndex<3>(data_format, 'C')] = in SetTfDimOrder()
507 data_.map_[GetTensorDimIndex<3>(data_format, 'N')] = in SetTfDimOrder()
512 data_.map_[GetTensorDimIndex<2>(data_format, 'W')] = MklDnnDims::Dim_W; in SetTfDimOrder()
513 data_.map_[GetTensorDimIndex<2>(data_format, 'H')] = MklDnnDims::Dim_H; in SetTfDimOrder()
514 data_.map_[GetTensorDimIndex<2>(data_format, 'C')] = MklDnnDims::Dim_C; in SetTfDimOrder()
515 data_.map_[GetTensorDimIndex<2>(data_format, 'N')] = MklDnnDims::Dim_N; in SetTfDimOrder()
557 return (IsMklBatchDim(GetTensorDimIndex<2>(data_format, 'N')) && in IsTensorInNCHWFormat()
[all …]
|
| D | tensor_format.h | 313 inline int32 GetTensorDimIndex(TensorFormat format, char dimension) { in GetTensorDimIndex() function
418 inline int32 GetTensorDimIndex(TensorFormat format, char dimension) { in GetTensorDimIndex() function
419 return GetTensorDimIndex<2>(format, dimension); in GetTensorDimIndex()
422 inline int32 GetTensorDimIndex(TensorFormat format, char dimension, in GetTensorDimIndex() function
425 ? GetTensorDimIndex<3>(format, dimension) in GetTensorDimIndex()
426 : GetTensorDimIndex<2>(format, dimension); in GetTensorDimIndex()
439 GetTensorDimIndex(tensor_format, dimension, dimension_attributes.size()); in GetTensorDim()
502 GetTensorDimIndex(tensor_format, dimension, explicit_paddings.size() / 2); in GetExplicitPaddingForDim()
|
| D | tensor_format_test.cc | 179 EXPECT_EQ(GetTensorDimIndex<num_spatial_dims>(format, 'N'), tdm.n()); in RunDimensionIndexesTest()
181 EXPECT_EQ(GetTensorDimIndex<num_spatial_dims>(format, 'C'), tdm.c()); in RunDimensionIndexesTest()
184 EXPECT_EQ(GetTensorDimIndex<num_spatial_dims>(format, '0' + i), in RunDimensionIndexesTest()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthtospace_op.cc | 83 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'N')); in Compute()
85 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'H')); in Compute()
87 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'W')); in Compute()
89 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'C')) * in Compute()
|
| D | spacetodepth_op.cc | 98 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'N')); in Compute()
100 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'H')); in Compute()
102 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'W')); in Compute()
104 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'C')) * in Compute()
|
| D | conv_grad_filter_ops.cc | 115 dilations[GetTensorDimIndex(data_format, 'H')] = row_dilation; in operator ()()
116 dilations[GetTensorDimIndex(data_format, 'W')] = col_dilation; in operator ()()
119 strides[GetTensorDimIndex(data_format, 'H')] = row_stride; in operator ()()
120 strides[GetTensorDimIndex(data_format, 'W')] = col_stride; in operator ()()
661 dilations[GetTensorDimIndex(data_format, 'H')] = row_dilation; in operator ()()
662 dilations[GetTensorDimIndex(data_format, 'W')] = col_dilation; in operator ()()
665 strides[GetTensorDimIndex(data_format, 'H')] = row_stride; in operator ()()
666 strides[GetTensorDimIndex(data_format, 'W')] = col_stride; in operator ()()
|
| D | conv_2d.h | 343 padding[GetTensorDimIndex<NDIMS - 2>(format, 'N')] = {0, 0};
345 padding[GetTensorDimIndex<NDIMS - 2>(format, '0' + i)] = {
348 padding[GetTensorDimIndex<NDIMS - 2>(format, 'C')] = {0, 0};
|
| D | conv_grad_input_ops.cc | 78 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator ()()
79 auto input_w = GetTensorDimIndex(data_format, 'W'); in operator ()()
|
| D | conv_grad_input_ops.h | 120 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator()
121 auto input_w = GetTensorDimIndex(data_format, 'W'); in operator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 770 return c->Dim(input_shape, GetTensorDimIndex<3>(FORMAT_NCHW, dimension)); in Conv3DShape()
1094 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in AvgPoolShape()
1096 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in AvgPoolShape()
1098 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in AvgPoolShape()
1100 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in AvgPoolShape()
1537 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in MaxPoolShapeImpl()
1539 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in MaxPoolShapeImpl()
1541 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in MaxPoolShapeImpl()
1543 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in MaxPoolShapeImpl()
1676 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in MaxPoolV2Shape()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2490 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in __anon42d741193702()
2492 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in __anon42d741193702()
2494 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in __anon42d741193702()
2496 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in __anon42d741193702()
2545 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in __anon42d741193802()
2547 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in __anon42d741193802()
2549 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in __anon42d741193802()
2551 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in __anon42d741193802()
|