Home
last modified time | relevance | path

Searched refs:GetTensorDimsFromSpatialDims (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/ops/
Dfused_conv2d_bias_activation_op.cc65 GetTensorDimsFromSpatialDims(num_spatial_dims, data_format); in __anon4e6e467d0102()
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h136 inline int GetTensorDimsFromSpatialDims(int num_spatial_dims, in GetTensorDimsFromSpatialDims() function
511 const int dims = GetTensorDimsFromSpatialDims(spatial.size(), format); in ShapeFromFormat()
Dtensor_format_test.cc175 int num_dims = GetTensorDimsFromSpatialDims(num_spatial_dims, format); in RunDimensionIndexesTest()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc332 const int num_dims = GetTensorDimsFromSpatialDims(spatial.size(), format); in MakeShapeFromFormat()
357 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format); in DimensionsFromShape()
381 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format); in ShapeFromDimensions()
432 const int rank = GetTensorDimsFromSpatialDims(num_spatial_dims, data_format); in Conv2DShapeImpl()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2325 GetTensorDimsFromSpatialDims(num_spatial_dims, data_format); in __anon7c94107b3702()
2379 GetTensorDimsFromSpatialDims(num_spatial_dims, data_format); in __anon7c94107b3802()