Searched refs:GetTensorSpatialDimIndex (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | extract_image_patches_op.cc | 78 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile()
111 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile()
140 const int64_t dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile()
|
| D | conv_op_helpers.cc | 124 int input_dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in CheckConvAttrs()
248 GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in MakeXlaForwardConvOp()
345 int64_t dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in MakeXlaBackpropInputConvOp()
461 int64_t dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in MakeXlaBackpropFilterConvOp()
|
| D | pooling_ops.cc | 153 GetTensorSpatialDimIndex(num_dims, data_format, spatial_dim); in XlaTensorFormat()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 226 inline int GetTensorSpatialDimIndex(int num_dims, TensorFormat format, in GetTensorSpatialDimIndex() function
538 dim_sizes[GetTensorSpatialDimIndex(dims, format, dim)] = dim_size; in ShapeFromFormat()
606 src_shape.dim_sizes())[GetTensorSpatialDimIndex( in ShapeFromFormat()
|
| D | tensor_format_test.cc | 183 EXPECT_EQ(GetTensorSpatialDimIndex(num_dims, format, i), tdm.spatial(i)); in RunDimensionIndexesTest()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.cc | 370 int64_t ifm_dim = GetTensorSpatialDimIndex( in getPaddingValuesFromPadType()
415 int64_t dim = GetTensorSpatialDimIndex(4, data_format_tf, in getPaddingValuesFromExplicitPadAttr()
444 int64_t ifm_dim = GetTensorSpatialDimIndex( in getTransposeConv2dPaddingValues()
446 int64_t ofm_dim = GetTensorSpatialDimIndex( in getTransposeConv2dPaddingValues()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.cc | 146 int image_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in ConvBackpropComputeDimensionsV2()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 515 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] = in MakeShapeFromFormat()
535 shape, GetTensorSpatialDimIndex(rank, format, spatial_dim_index)); in DimensionsFromShape()
563 out_dims[tensorflow::GetTensorSpatialDimIndex( in ShapeFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1173 const int64_t dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in matchAndRewriteDynamicConv()
1323 const int64_t dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in matchAndRewrite()
2814 int dim = tensorflow::GetTensorSpatialDimIndex(num_dims, data_format, i); in matchAndRewrite()
4951 GetTensorSpatialDimIndex(num_dims, data_format, i); in matchAndRewrite()
5180 tensorflow::GetTensorSpatialDimIndex(num_dims, data_format, i); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 108 int64 dim = GetTensorSpatialDimIndex(4, data_format_tf, i)
1052 int64 ifm_dim = GetTensorSpatialDimIndex(4, data_format, i);
1053 int64 ofm_dim = GetTensorSpatialDimIndex(4, data_format, i);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1774 const int64_t dim = GetTensorSpatialDimIndex(num_dims, format, i); in inferConvReturnTypeComponents()
|