Searched refs:GetTensorBatchDimIndex (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 173 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in CheckConvAttrs()
269 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp()
329 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp()
438 int n_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp()
|
| D | extract_image_patches_op.cc | 54 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in Compile()
|
| D | pooling_ops.cc | 138 int batch_dimension = GetTensorBatchDimIndex(num_dims, data_format); in XlaTensorFormat()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | padding.cc | 55 const int32 batch_index = GetTensorBatchDimIndex(num_dims, data_format); in CheckValidPadding()
|
| D | tensor_format.h | 162 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) { in GetTensorBatchDimIndex() function
513 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/core/kernels/ |
| D | conv_grad_ops.cc | 118 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 334 dims_actual[GetTensorBatchDimIndex(num_dims, format)] = context->MakeDim(N); in MakeShapeFromFormat()
359 *batch_dim = context->Dim(shape, GetTensorBatchDimIndex(rank, format)); in DimensionsFromShape()
385 out_dims[tensorflow::GetTensorBatchDimIndex(rank, format)] = batch_dim; in ShapeFromDimensions()
|