Searched refs:depth_idx (Results 1 – 1 of 1) sorted by relevance
164 int depth_idx, batch_idx, spatial_idx; in GetDimIndices() local167 depth_idx = data_dims - 1; in GetDimIndices()173 depth_idx = data_dims - 2; in GetDimIndices()179 depth_idx = data_dims - 1; in GetDimIndices()186 depth_idx = 1; in GetDimIndices()195 return std::make_tuple(depth_idx, batch_idx, spatial_idx); in GetDimIndices()