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 depth_index = GetTensorFeatureDimIndex(num_dims, data_format); in CheckValidPadding()
Dtensor_format.h193 inline int GetTensorFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorFeatureDimIndex() function
540 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()
211 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.cc86 GetTensorFeatureDimIndex(input_shape.dims(), data_format_); in CompileImpl()
269 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.cc141 int feature_dimension = GetTensorFeatureDimIndex(num_dims, data_format); in XlaTensorFormat()
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc124 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc501 int outer_c_index = GetTensorFeatureDimIndex(num_dims, format); in MakeShapeFromFormat()
534 *filter_dim = context->Dim(shape, GetTensorFeatureDimIndex(rank, format)); in DimensionsFromShape()
565 &out_dims[tensorflow::GetTensorFeatureDimIndex(rank, format)])); in ShapeFromDimensions()
568 out_dims[tensorflow::GetTensorFeatureDimIndex(rank, format)] = filter_dim; in ShapeFromDimensions()
1152 int channel_dim_index = GetTensorFeatureDimIndex(rank, data_format); in FusedBatchNormShape()
1192 int channel_dim_index = GetTensorFeatureDimIndex(4, data_format); in FusedBatchNormExShape()
1222 int channel_dim_index = GetTensorFeatureDimIndex(rank, data_format); in FusedBatchNormGradShape()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc477 tensorflow::GetTensorFeatureDimIndex(value_ty.getRank(), format); in Verify()
1526 int idx = tensorflow::GetTensorFeatureDimIndex(num_dims, format); in Verify()
1641 return_shape[GetTensorFeatureDimIndex(num_dims, format)] = in inferConvReturnTypes()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc104 int64_t feature_dim = GetTensorFeatureDimIndex(rank, format); in GetBiasAddGradReductionIndices()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1099 builder->getI64IntegerAttr(GetTensorFeatureDimIndex(num_dims, format)); in GetConvDimensionNumbersAttr()
1240 GetDimSize(input_ty, GetTensorFeatureDimIndex(num_dims, data_format)); in matchAndRewrite()
4190 tensorflow::GetTensorFeatureDimIndex(num_dims, data_format); in matchAndRewrite()
4355 tensorflow::GetTensorFeatureDimIndex(num_dims, data_format); in matchAndRewrite()