Searched refs:broadcast_starts (Results 1 – 2 of 2) sorted by relevance
48 bool broadcast_starts = starts_in.shape().dims() == 0; in Compute() local55 if (!broadcast_starts) in_sizes.push_back(starts_in.shape().dim_size(0)); in Compute()77 T start = broadcast_starts ? starts(0) : starts(row); in Compute()94 T value = broadcast_starts ? starts(0) : starts(row); in Compute()
67 const bool broadcast_starts = NumElements(&input_starts) == 1; in EvalT() local74 if (!broadcast_starts) in_sizes.push_back(input_starts.dims->data[0]); in EvalT()102 const T start = broadcast_starts ? starts[0] : starts[row]; in EvalT()124 T value = broadcast_starts ? starts[0] : starts[row]; in EvalT()