Searched refs:in_sizes (Results 1 – 2 of 2) sorted by relevance
54 std::vector<int> in_sizes; in Compute() local55 if (!broadcast_starts) in_sizes.push_back(starts_in.shape().dim_size(0)); in Compute()56 if (!broadcast_limits) in_sizes.push_back(limits_in.shape().dim_size(0)); in Compute()57 if (!broadcast_deltas) in_sizes.push_back(deltas_in.shape().dim_size(0)); in Compute()58 for (int i = 1; i < in_sizes.size(); ++i) { in Compute()59 OP_REQUIRES(context, in_sizes[i] == in_sizes[i - 1], in Compute()63 SPLITS_TYPE nrows = in_sizes.empty() ? 1 : in_sizes[0]; in Compute()
73 std::vector<int> in_sizes; in EvalT() local74 if (!broadcast_starts) in_sizes.push_back(input_starts.dims->data[0]); in EvalT()75 if (!broadcast_limits) in_sizes.push_back(input_limits.dims->data[0]); in EvalT()76 if (!broadcast_deltas) in_sizes.push_back(input_deltas.dims->data[0]); in EvalT()77 if (std::adjacent_find(std::begin(in_sizes), std::end(in_sizes), in EvalT()78 std::not_equal_to<>()) != std::end(in_sizes)) { in EvalT()86 const SPLITS_TYPE nrows = in_sizes.empty() ? 1 : in_sizes.front(); in EvalT()