Searched defs:input_num (Results 1 – 8 of 8) sorted by relevance
70 for (int input_num = 0; input_num < indices_input.size(); input_num++) { in Compile() local110 for (int input_num = 0; input_num < indices.size(); input_num++) { in Compile() local136 for (int input_num = 0; input_num < indices.size(); input_num++) { in Compile() local153 for (int input_num = 0; input_num < indices.size(); input_num++) { in Compile() local
203 int input_num = then_result.input_mapping[i] + 1; in Compile() local
215 int input_num = branch_results[0].input_mapping[i] + 1; in Compile() local
468 int input_num = body.input_mapping[i]; in Compile() local
104 for (int input_num = 0; input_num < indices_inputs->size(); input_num++) { in CheckArgsAndAllocateResult() local254 auto OnInputNumber = [&](int input_num) { in Compute()291 for (int input_num = 0; input_num < indices_inputs.size(); in Compute() local299 for (int input_num = first; input_num < last; ++input_num) { in Compute() local306 for (int input_num = 0; input_num < indices_inputs.size(); in Compute() local
139 for (int input_num = 0; input_num < c->num_inputs(); input_num++) { in Compute() local
269 int64 input_num = xla_args.size(); in ConvertGraphToXla() local
449 for (int64 input_num = 0; input_num < ctx->num_inputs(); ++input_num) { in BuildXlaCompilerArguments() local