Searched refs:num_total_dims (Results 1 – 1 of 1) sorted by relevance
423 int num_total_dims) { in GetTensorDimIndex() argument424 int32 index = (GetTensorSpatialDims(num_total_dims, format) == 3) in GetTensorDimIndex()427 CHECK(index >= 0 && index < num_total_dims) // Crash OK. in GetTensorDimIndex()