Home
last modified time | relevance | path

Searched refs:GetTensorFeatureDimIndex (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dpadding.cc53 const int32_t depth_index = GetTensorFeatureDimIndex(num_dims, data_format); in CheckValidPadding()
Dtensor_format.h193 inline int GetTensorFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorFeatureDimIndex() function
541 int feature_index = GetTensorFeatureDimIndex(dims, format); in ShapeFromFormat()
Dtensor_format_test.cc180 EXPECT_EQ(GetTensorFeatureDimIndex(num_dims, format), tdm.c()); in RunDimensionIndexesTest()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc107 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in CheckConvAttrs()
210 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp()
301 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp()
431 int c_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp()
Dbatch_norm_op.cc90 GetTensorFeatureDimIndex(input_shape.dims(), data_format_); in CompileImpl()
273 GetTensorFeatureDimIndex(input_dims, data_format_); in Compile()
Ddepthtospace_op.cc69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile()
Dspacetodepth_op.cc69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile()
Dextract_image_patches_op.cc62 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in Compile()
Dpooling_ops.cc149 int feature_dimension = GetTensorFeatureDimIndex(num_dims, data_format); in XlaTensorFormat()
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc125 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc504 int outer_c_index = GetTensorFeatureDimIndex(num_dims, format); in MakeShapeFromFormat()
538 *filter_dim = context->Dim(shape, GetTensorFeatureDimIndex(rank, format)); in DimensionsFromShape()
573 &out_dims[tensorflow::GetTensorFeatureDimIndex(rank, format)])); in ShapeFromDimensions()
576 out_dims[tensorflow::GetTensorFeatureDimIndex(rank, format)] = filter_dim; in ShapeFromDimensions()
1207 int channel_dim_index = GetTensorFeatureDimIndex(rank, data_format); in FusedBatchNormShape()
1247 int channel_dim_index = GetTensorFeatureDimIndex(4, data_format); in FusedBatchNormExShape()
1277 int channel_dim_index = GetTensorFeatureDimIndex(rank, data_format); in FusedBatchNormGradShape()
1323 int channel_dim_index = GetTensorFeatureDimIndex(rank, data_format); in FusedBatchNormGradExShape()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc527 tensorflow::GetTensorFeatureDimIndex(value_ty.getRank(), format); in verify()
1681 int idx = tensorflow::GetTensorFeatureDimIndex(num_dims, format); in Verify()
1765 return_shape[GetTensorFeatureDimIndex(num_dims, format)] = in inferConvReturnTypeComponents()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc79 return GetTensorFeatureDimIndex(input_ty.getRank(), format); in GetFeatureDimension()
991 int64_t feature_dim = GetTensorFeatureDimIndex(num_dims, format); in GetConvDimensionNumbersAttr()
1207 GetDimSize(input_ty, GetTensorFeatureDimIndex(num_dims, data_format)); in matchAndRewriteDynamicConv()
1362 GetDimSize(input_ty, GetTensorFeatureDimIndex(num_dims, data_format)); in matchAndRewrite()
4999 tensorflow::GetTensorFeatureDimIndex(num_dims, data_format); in matchAndRewrite()
5162 tensorflow::GetTensorFeatureDimIndex(num_dims, data_format); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc140 int64_t feature_dim = GetTensorFeatureDimIndex(rank, format); in GetBiasAddGradReductionIndices()