Home
last modified time | relevance | path

Searched refs:GetTensorSpatialDims (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dtensor_format.h111 inline int GetTensorSpatialDims(int num_dims, TensorFormat format) { in GetTensorSpatialDims() function
218 spatial_dim < GetTensorSpatialDims(num_dims, format)) in GetTensorSpatialDimIndex()
413 int32 index = (GetTensorSpatialDims(num_total_dims, format) == 3) in GetTensorDimIndex()
588 GetTensorSpatialDims(src_shape.dims(), src_format); in ShapeFromFormat()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddepthtospace_op.cc70 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format); in Compile()
Dspacetodepth_op.cc70 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format); in Compile()
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.cc55 GetTensorSpatialDims(tensor_in_shape.dims(), data_format) == 2, in PoolParameters()