Home
last modified time | relevance | path

Searched refs:GetTensorInnerFeatureDimIndex (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dtensor_format.h200 inline int GetTensorInnerFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerFeatureDimIndex() function
532 dim_sizes[GetTensorInnerFeatureDimIndex(dims, format)] = 4; in ShapeFromFormat()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc320 shape_handle, GetTensorInnerFeatureDimIndex(num_dims, tensor_format)); in CheckFormatConstraintsOnShape()
338 dims_actual[GetTensorInnerFeatureDimIndex(num_dims, format)] = in MakeShapeFromFormat()
371 context->Dim(shape, GetTensorInnerFeatureDimIndex(rank, format)), in DimensionsFromShape()
399 out_dims[GetTensorInnerFeatureDimIndex(rank, format)] = context->MakeDim(4); in ShapeFromDimensions()