Searched refs:GetTensorSpatialDimIndex (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | extract_image_patches_op.cc | 71 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile() 104 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile() 134 const int64 dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile()
|
D | conv_op_helpers.cc | 190 int input_dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in CheckConvAttrs() 299 const int64 dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in MakeXlaForwardConvOp() 368 int64 dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in MakeXlaBackpropInputConvOp() 474 int64 dim = GetTensorSpatialDimIndex(num_dims, attrs.data_format, i); in MakeXlaBackpropFilterConvOp()
|
D | pooling_ops.cc | 143 GetTensorSpatialDimIndex(num_dims, data_format, spatial_dim); in XlaTensorFormat()
|
/external/tensorflow/tensorflow/core/util/ |
D | tensor_format.h | 215 inline int GetTensorSpatialDimIndex(int num_dims, TensorFormat format, in GetTensorSpatialDimIndex() function 524 dim_sizes[GetTensorSpatialDimIndex(dims, format, dim)] = dim_size; in ShapeFromFormat() 592 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_ops.cc | 145 int image_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in ConvBackpropComputeDimensionsV2()
|
/external/tensorflow/tensorflow/core/framework/ |
D | common_shape_fns.cc | 345 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] = in MakeShapeFromFormat() 364 shape, GetTensorSpatialDimIndex(rank, format, spatial_dim_index)); in DimensionsFromShape() 389 out_dims[tensorflow::GetTensorSpatialDimIndex( in ShapeFromDimensions()
|