Searched refs:num_total_dims (Results 1 – 2 of 2) sorted by relevance
412 int num_total_dims) { in GetTensorDimIndex() argument413 int32 index = (GetTensorSpatialDims(num_total_dims, format) == 3) in GetTensorDimIndex()416 CHECK(index >= 0 && index < num_total_dims) // Crash OK. in GetTensorDimIndex()
1016 num_total_dims = filter_shape.ndims1017 if num_total_dims is None:1018 num_total_dims = input_shape.ndims1019 if num_total_dims is None:1022 num_spatial_dims = num_total_dims - 2