Home
last modified time | relevance | path

Searched refs:GetTensorDimIndex (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_util.h230 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'W')] = MklDims::W; in SetTfDimOrder()
231 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'H')] = MklDims::H; in SetTfDimOrder()
232 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'C')] = MklDims::C; in SetTfDimOrder()
233 tf_to_mkl_dim_map_[GetTensorDimIndex<2>(data_format, 'N')] = MklDims::N; in SetTfDimOrder()
268 return (IsMklBatchDim(GetTensorDimIndex<2>(data_format, 'N')) && in IsTensorInNCHWFormat()
269 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNCHWFormat()
270 IsMklHeightDim(GetTensorDimIndex<2>(data_format, 'H')) && in IsTensorInNCHWFormat()
271 IsMklWidthDim(GetTensorDimIndex<2>(data_format, 'W'))); in IsTensorInNCHWFormat()
278 return (IsMklBatchDim(GetTensorDimIndex<2>(data_format, 'N')) && in IsTensorInNHWCFormat()
279 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNHWCFormat()
[all …]
Dtensor_format.h302 inline int32 GetTensorDimIndex(TensorFormat format, char dimension) { in GetTensorDimIndex() function
407 inline int32 GetTensorDimIndex(TensorFormat format, char dimension) { in GetTensorDimIndex() function
408 return GetTensorDimIndex<2>(format, dimension); in GetTensorDimIndex()
411 inline int32 GetTensorDimIndex(TensorFormat format, char dimension, in GetTensorDimIndex() function
414 ? GetTensorDimIndex<3>(format, dimension) in GetTensorDimIndex()
415 : GetTensorDimIndex<2>(format, dimension); in GetTensorDimIndex()
428 GetTensorDimIndex(tensor_format, dimension, dimension_attributes.size()); in GetTensorDim()
489 GetTensorDimIndex(tensor_format, dimension, explicit_paddings.size() / 2); in GetExplicitPaddingForDim()
Dtensor_format_test.cc179 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/
Ddepthtospace_op.cc83 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()
Dspacetodepth_op.cc83 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()
Dconv_2d.h255 padding[GetTensorDimIndex<NDIMS - 2>(format, 'N')] = {0, 0};
257 padding[GetTensorDimIndex<NDIMS - 2>(format, '0' + i)] = {
260 padding[GetTensorDimIndex<NDIMS - 2>(format, 'C')] = {0, 0};
Dconv_grad_filter_ops.cc554 dilations[GetTensorDimIndex(data_format, 'H')] = row_dilation; in operator ()()
555 dilations[GetTensorDimIndex(data_format, 'W')] = col_dilation; in operator ()()
558 strides[GetTensorDimIndex(data_format, 'H')] = row_stride; in operator ()()
559 strides[GetTensorDimIndex(data_format, 'W')] = col_stride; in operator ()()
Dconv_grad_input_ops.cc692 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator ()()
693 auto input_w = GetTensorDimIndex(data_format, 'W'); in operator ()()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc587 return c->Dim(input_shape, GetTensorDimIndex<3>(FORMAT_NCHW, dimension)); in Conv3DShape()
760 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in AvgPoolShape()
762 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in AvgPoolShape()
764 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in AvgPoolShape()
766 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in AvgPoolShape()
912 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in MaxPoolShape()
914 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in MaxPoolShape()
916 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in MaxPoolShape()
918 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in MaxPoolShape()
1014 input_shape, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in MaxPoolV2Shape()
[all …]
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2333 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in __anondb9326b23702()
2335 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in __anondb9326b23702()
2337 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in __anondb9326b23702()
2339 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in __anondb9326b23702()
2388 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'N')); in __anondb9326b23802()
2390 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'H')); in __anondb9326b23802()
2392 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'W')); in __anondb9326b23802()
2394 c->Dim(input, GetTensorDimIndex<num_spatial_dims>(data_format, 'C')); in __anondb9326b23802()