Searched refs:GetTensorInnerFeatureDimIndex (Results 1 – 2 of 2) sorted by relevance
200 inline int GetTensorInnerFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerFeatureDimIndex() function532 dim_sizes[GetTensorInnerFeatureDimIndex(dims, format)] = 4; in ShapeFromFormat()
320 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()