Searched defs:input_num (Results 1 – 10 of 10) 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
287 int input_num = branch_results[0].input_mapping[i] + 1; in Compile() local
275 int input_num = then_result.input_mapping[i] + 1; in Compile() local
488 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() local255 auto OnInputNumber = [&](int input_num) { in Compute()292 for (int input_num = 0; input_num < indices_inputs.size(); in Compute() local300 for (int input_num = first; input_num < last; ++input_num) { in Compute() local307 for (int input_num = 0; input_num < indices_inputs.size(); in Compute() local
54 for (int input_num = 0; input_num < TF_NumInputs(ctx); ++input_num) { in MergeSummaryOp_Compute() local
61 int input_num = -1; in ApplyToNode() local
650 for (auto input_num : op.inputs) { in ConvertOp() local1063 for (auto input_num : op->inputs) { in ConvertSubgraph() local
610 for (int64 input_num = 0; input_num < inputs.size(); ++input_num) { in BuildXlaCompilerArguments() local
2266 int64 input_num = in BuildDynamicShapeNodes() local3121 int64 input_num = params_info.IsPerReplicaArg(orig_arg_num) in BuildExecuteNodes() local3192 int64 input_num = params_info.FirstBroadcastArgFromHost() + in BuildExecuteNodes() local