Searched refs:GetFilterDimIndex (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format_test.cc | 195 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()
|
| D | tensor_format.h | 370 inline int GetFilterDimIndex(FilterTensorFormat filter_tensor_format, in GetFilterDimIndex() function
439 ? GetFilterDimIndex<3>(filter_tensor_format, dimension) in GetFilterDim()
440 : GetFilterDimIndex<2>(filter_tensor_format, dimension); in GetFilterDim()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
| D | fused_conv2d_bias_activation_op.cc | 71 GetFilterDimIndex<num_spatial_dims>(filter_format, 'O')); in __anon4e6e467d0102()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 477 filter_shape, GetFilterDimIndex<num_spatial_dims>(filter_format, 'O')); in Conv2DShapeImpl()
479 filter_shape, GetFilterDimIndex<num_spatial_dims>(filter_format, 'H')); in Conv2DShapeImpl()
481 filter_shape, GetFilterDimIndex<num_spatial_dims>(filter_format, 'W')); in Conv2DShapeImpl()
486 GetFilterDimIndex<num_spatial_dims>(filter_format, 'I')), in Conv2DShapeImpl()
492 filter_shape, GetFilterDimIndex<num_spatial_dims>(filter_format, 'I')); in Conv2DShapeImpl()
|