Searched refs:num_classes_raw (Results 1 – 2 of 2) sorted by relevance
195 const int64 num_classes_raw = inputs_shape.dim_size(2); in Compute() local197 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()199 const int num_classes = static_cast<const int>(num_classes_raw); in Compute()286 const int64 num_classes_raw = inputs_shape.dim_size(2); in Compute() local288 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()290 const int num_classes = static_cast<const int>(num_classes_raw); in Compute()
71 const int64 num_classes_raw = inputs_shape.dim_size(2); in Compute() local73 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()75 const int num_classes = static_cast<const int>(num_classes_raw); in Compute()