Home
last modified time | relevance | path

Searched refs:GetTensorSpatialDimIndex (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dextract_image_patches_op.cc71 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()
Dconv_op_helpers.cc190 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()
Dpooling_ops.cc143 GetTensorSpatialDimIndex(num_dims, data_format, spatial_dim); in XlaTensorFormat()
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h215 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()
Dtensor_format_test.cc183 EXPECT_EQ(GetTensorSpatialDimIndex(num_dims, format, i), tdm.spatial(i)); in RunDimensionIndexesTest()
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_ops.cc145 int image_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in ConvBackpropComputeDimensionsV2()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc345 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()