Lines Matching refs:input_num
104 for (int input_num = 0; input_num < indices_inputs->size(); input_num++) {
105 const Tensor& indices = (*indices_inputs)[input_num];
106 const Tensor& data = (*data_inputs)[input_num];
109 errors::InvalidArgument("data[", input_num,
111 " does not start with indices[", input_num,
114 c, input_num == 0 || SameExtraShape(data0, indices0, data, indices),
116 "Need data[0].shape[", indices0.dims(), ":] = data[", input_num,
119 ", data[", input_num, "].shape = ", data.shape().DebugString(),
121 ", indices[", input_num,
242 auto OnInputNumber = [&](int input_num) {
243 const Tensor& indices = indices_inputs[input_num];
245 const Tensor& data = data_inputs[input_num];
279 for (int input_num = 0; input_num < indices_inputs.size();
280 ++input_num) {
281 total_indices_size += indices_inputs[input_num].NumElements();
287 for (int input_num = first; input_num < last; ++input_num) {
288 OnInputNumber(input_num);
294 for (int input_num = 0; input_num < indices_inputs.size();
295 input_num++) {
296 OnInputNumber(input_num);
