| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 496 data_.map_[GetTensorDimIndex<3>(data_format, '0')] = in SetTfDimOrder()
498 data_.map_[GetTensorDimIndex<3>(data_format, '1')] = in SetTfDimOrder()
500 data_.map_[GetTensorDimIndex<3>(data_format, '2')] = in SetTfDimOrder()
502 data_.map_[GetTensorDimIndex<3>(data_format, 'C')] = in SetTfDimOrder()
504 data_.map_[GetTensorDimIndex<3>(data_format, 'N')] = in SetTfDimOrder()
509 data_.map_[GetTensorDimIndex<2>(data_format, 'W')] = MklDnnDims::Dim_W; in SetTfDimOrder()
510 data_.map_[GetTensorDimIndex<2>(data_format, 'H')] = MklDnnDims::Dim_H; in SetTfDimOrder()
511 data_.map_[GetTensorDimIndex<2>(data_format, 'C')] = MklDnnDims::Dim_C; in SetTfDimOrder()
512 data_.map_[GetTensorDimIndex<2>(data_format, 'N')] = MklDnnDims::Dim_N; in SetTfDimOrder()
554 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 | 119 dilations[GetTensorDimIndex(data_format, 'H')] = row_dilation; in operator ()()
120 dilations[GetTensorDimIndex(data_format, 'W')] = col_dilation; in operator ()()
123 strides[GetTensorDimIndex(data_format, 'H')] = row_stride; in operator ()()
124 strides[GetTensorDimIndex(data_format, 'W')] = col_stride; in operator ()()
680 dilations[GetTensorDimIndex(data_format, 'H')] = row_dilation; in operator ()()
681 dilations[GetTensorDimIndex(data_format, 'W')] = col_dilation; in operator ()()
684 strides[GetTensorDimIndex(data_format, 'H')] = row_stride; in operator ()()
685 strides[GetTensorDimIndex(data_format, 'W')] = col_stride; in operator ()()
|
| D | conv_2d.h | 435 padding[GetTensorDimIndex<NDIMS - 2>(format, 'N')] = {0, 0};
437 padding[GetTensorDimIndex<NDIMS - 2>(format, '0' + i)] = {
440 padding[GetTensorDimIndex<NDIMS - 2>(format, 'C')] = {0, 0};
|
| D | conv_grad_input_ops.cc | 85 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator ()()
86 auto input_w = GetTensorDimIndex(data_format, 'W'); in operator ()()
|
| D | conv_grad_input_ops.h | 121 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator()
122 auto input_w = GetTensorDimIndex(data_format, 'W'); in operator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 802 return c->Dim(input_shape, GetTensorDimIndex<3>(FORMAT_NCHW, dimension)); in Conv3DShape()
1149 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in AvgPoolShape()
1151 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in AvgPoolShape()
1153 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in AvgPoolShape()
1155 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in AvgPoolShape()
1628 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in MaxPoolShapeImpl()
1630 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in MaxPoolShapeImpl()
1632 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in MaxPoolShapeImpl()
1634 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in MaxPoolShapeImpl()
1767 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in MaxPoolV2Shape()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2523 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in __anon847f0b683702()
2525 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in __anon847f0b683702()
2527 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in __anon847f0b683702()
2529 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in __anon847f0b683702()
2578 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in __anon847f0b683802()
2580 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in __anon847f0b683802()
2582 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in __anon847f0b683802()
2584 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in __anon847f0b683802()
|