Searched refs:GetTensorBatchDimIndex (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | padding.cc | 52 const int32 batch_index = GetTensorBatchDimIndex(num_dims, data_format); in CheckValidPadding()
|
| D | tensor_format.h | 173 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) { in GetTensorBatchDimIndex() function
526 dim_sizes[GetTensorBatchDimIndex(dims, format)] = N; in ShapeFromFormat()
|
| D | tensor_format_test.cc | 178 EXPECT_EQ(GetTensorBatchDimIndex(num_dims, format), tdm.n()); in RunDimensionIndexesTest()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 106 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in CheckConvAttrs()
210 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()
|
| D | extract_image_patches_op.cc | 61 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in Compile()
|
| D | pooling_ops.cc | 140 int batch_dimension = GetTensorBatchDimIndex(num_dims, data_format); in XlaTensorFormat()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.cc | 114 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 500 dims_actual[GetTensorBatchDimIndex(num_dims, format)] = context->MakeDim(N); in MakeShapeFromFormat()
526 *batch_dim = context->Dim(shape, GetTensorBatchDimIndex(rank, format)); in DimensionsFromShape()
552 out_dims[tensorflow::GetTensorBatchDimIndex(rank, format)] = batch_dim; in ShapeFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1637 return_shape[GetTensorBatchDimIndex(num_dims, format)] = in inferConvReturnTypes()
1638 input_ty.getDimSize(GetTensorBatchDimIndex(num_dims, format)); in inferConvReturnTypes()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1097 builder->getI64IntegerAttr(GetTensorBatchDimIndex(num_dims, format)); in GetConvDimensionNumbersAttr()
4231 tensorflow::GetTensorBatchDimIndex(num_dims, data_format); in matchAndRewrite()
4438 tensorflow::GetTensorBatchDimIndex(num_dims, data_format); in matchAndRewrite()
|