Searched refs:broadcast_limits (Results 1 – 2 of 2) sorted by relevance
49 bool broadcast_limits = limits_in.shape().dims() == 0; in Compute() local56 if (!broadcast_limits) in_sizes.push_back(limits_in.shape().dim_size(0)); in Compute()78 T limit = broadcast_limits ? limits(0) : limits(row); in Compute()
68 const bool broadcast_limits = NumElements(&input_limits) == 1; in EvalT() local75 if (!broadcast_limits) in_sizes.push_back(input_limits.dims->data[0]); in EvalT()103 const T limit = broadcast_limits ? limits[0] : limits[row]; in EvalT()