Searched refs:GetTensorBatchDimIndex (Results 1 – 10 of 10) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | conv_op_helpers.cc | 106 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in CheckConvAttrs() 209 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp() 300 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp() 430 int n_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp()
|
H A D | extract_image_patches_op.cc | 61 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in Compile()
|
H A D | pooling_ops.cc | 148 int batch_dimension = GetTensorBatchDimIndex(num_dims, data_format); in XlaTensorFormat()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
H A D | tensor_format.h | 173 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) { in GetTensorBatchDimIndex() function 527 dim_sizes[GetTensorBatchDimIndex(dims, format)] = N; in ShapeFromFormat()
|
H A D | padding.cc | 52 const int32_t batch_index = GetTensorBatchDimIndex(num_dims, data_format); in CheckValidPadding()
|
H A D | tensor_format_test.cc | 178 EXPECT_EQ(GetTensorBatchDimIndex(num_dims, format), tdm.n()); in RunDimensionIndexesTest()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | conv_grad_shape_utils.cc | 115 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | common_shape_fns.cc | 503 dims_actual[GetTensorBatchDimIndex(num_dims, format)] = context->MakeDim(N); in MakeShapeFromFormat() 530 *batch_dim = context->Dim(shape, GetTensorBatchDimIndex(rank, format)); in DimensionsFromShape() 559 out_dims[tensorflow::GetTensorBatchDimIndex(rank, format)] = batch_dim; in ShapeFromDimensions()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
H A D | tf_ops_a_m.cc | 1761 return_shape[GetTensorBatchDimIndex(num_dims, format)] = in inferConvReturnTypeComponents() 1762 input_ty.getDimSize(GetTensorBatchDimIndex(num_dims, format)); in inferConvReturnTypeComponents()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
H A D | legalize_tf.cc | 990 int64_t batch_dim = GetTensorBatchDimIndex(num_dims, format); in GetConvDimensionNumbersAttr() 4867 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in matchAndRewrite() 5248 tensorflow::GetTensorBatchDimIndex(num_dims, data_format); in matchAndRewrite()
|