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 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 const int64 dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in MakeXlaForwardConvOp()
345 int64 dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in MakeXlaBackpropInputConvOp()
461 int64 dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in MakeXlaBackpropFilterConvOp()
|
| D | pooling_ops.cc | 145 GetTensorSpatialDimIndex(num_dims, data_format, spatial_dim); in XlaTensorFormat()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.cc | 248 int64_t ifm_dim = GetTensorSpatialDimIndex( in getPaddingValuesFromPadType()
290 int64_t dim = GetTensorSpatialDimIndex(4, data_format_tf, in getPaddingValuesFromExplicitPadAttr()
318 int64_t ifm_dim = GetTensorSpatialDimIndex( in getTransposeConv2dPaddingValues()
320 int64_t ofm_dim = GetTensorSpatialDimIndex( in getTransposeConv2dPaddingValues()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 226 inline int GetTensorSpatialDimIndex(int num_dims, TensorFormat format, in GetTensorSpatialDimIndex() function
537 dim_sizes[GetTensorSpatialDimIndex(dims, format, dim)] = dim_size; in ShapeFromFormat()
605 gtl::ArraySlice<int64>(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/core/kernels/ |
| D | conv_grad_shape_utils.cc | 141 int image_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in ConvBackpropComputeDimensionsV2()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 512 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] = in MakeShapeFromFormat()
531 shape, GetTensorSpatialDimIndex(rank, format, spatial_dim_index)); in DimensionsFromShape()
556 out_dims[tensorflow::GetTensorSpatialDimIndex( in ShapeFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 145 int64_t dim = GetTensorSpatialDimIndex(4, data_format_tf, i)
1041 int64_t ifm_dim = GetTensorSpatialDimIndex(4, data_format, i);
1042 int64_t ofm_dim = GetTensorSpatialDimIndex(4, data_format, i);
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1203 const int64_t dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in matchAndRewrite()
2416 int dim = tensorflow::GetTensorSpatialDimIndex(num_dims, data_format, i); in matchAndRewrite()
4172 const int64_t dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in matchAndRewrite()
4373 tensorflow::GetTensorSpatialDimIndex(num_dims, data_format, i); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1650 const int64_t dim = GetTensorSpatialDimIndex(num_dims, format, i); in inferConvReturnTypes()
|