Home
last modified time | relevance | path

Searched refs:GetFilterDimIndex (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dtensor_format_test.cc195 EXPECT_EQ(GetFilterDimIndex<num_spatial_dims>(format, 'O'), fdm.o()); in RunDimensionIndexesTest()
197 EXPECT_EQ(GetFilterDimIndex<num_spatial_dims>(format, 'I'), fdm.i()); in RunDimensionIndexesTest()
201 EXPECT_EQ(GetFilterDimIndex<num_spatial_dims>(format, '0' + i), in RunDimensionIndexesTest()
Dtensor_format.h381 inline int GetFilterDimIndex(FilterTensorFormat filter_tensor_format, in GetFilterDimIndex() function
450 ? GetFilterDimIndex<3>(filter_tensor_format, dimension) in GetFilterDim()
451 : GetFilterDimIndex<2>(filter_tensor_format, dimension); in GetFilterDim()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc644 filter_shape, GetFilterDimIndex<num_spatial_dims>(filter_format, 'O')); in Conv2DShapeImpl()
646 filter_shape, GetFilterDimIndex<num_spatial_dims>(filter_format, 'H')); in Conv2DShapeImpl()
648 filter_shape, GetFilterDimIndex<num_spatial_dims>(filter_format, 'W')); in Conv2DShapeImpl()
653 GetFilterDimIndex<num_spatial_dims>(filter_format, 'I')), in Conv2DShapeImpl()
659 filter_shape, GetFilterDimIndex<num_spatial_dims>(filter_format, 'I')); in Conv2DShapeImpl()