Searched refs:GetTensorFeatureDimIndex (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 174 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in CheckConvAttrs()
270 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp()
330 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp()
439 int c_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp()
|
| D | batch_norm_op.cc | 54 GetTensorFeatureDimIndex(input_shape.dims(), data_format_); in Compile()
143 GetTensorFeatureDimIndex(input_dims, data_format_); in Compile()
|
| D | depthtospace_op.cc | 69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile()
|
| D | spacetodepth_op.cc | 69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile()
|
| D | extract_image_patches_op.cc | 55 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in Compile()
|
| D | pooling_ops.cc | 139 int feature_dimension = GetTensorFeatureDimIndex(num_dims, data_format); in XlaTensorFormat()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | padding.cc | 56 const int32 depth_index = GetTensorFeatureDimIndex(num_dims, data_format); in CheckValidPadding()
|
| D | tensor_format.h | 182 inline int GetTensorFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorFeatureDimIndex() function
527 int feature_index = GetTensorFeatureDimIndex(dims, format); in ShapeFromFormat()
|
| D | tensor_format_test.cc | 180 EXPECT_EQ(GetTensorFeatureDimIndex(num_dims, format), tdm.c()); in RunDimensionIndexesTest()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops.cc | 128 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 335 int outer_c_index = GetTensorFeatureDimIndex(num_dims, format); in MakeShapeFromFormat()
367 *filter_dim = context->Dim(shape, GetTensorFeatureDimIndex(rank, format)); in DimensionsFromShape()
398 &out_dims[tensorflow::GetTensorFeatureDimIndex(rank, format)])); in ShapeFromDimensions()
401 out_dims[tensorflow::GetTensorFeatureDimIndex(rank, format)] = filter_dim; in ShapeFromDimensions()
803 int channel_dim_index = GetTensorFeatureDimIndex(4, data_format); in FusedBatchNormShape()
839 int channel_dim_index = GetTensorFeatureDimIndex(4, data_format); in FusedBatchNormGradShape()
|