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() local120 for (int input_num = 0; input_num < indices.size(); input_num++) { in Compile() local146 for (int input_num = 0; input_num < indices.size(); input_num++) { in Compile() local163 for (int input_num = 0; input_num < indices.size(); input_num++) { in Compile() local
293 int input_num = then_result.input_mapping[i] + 1; in Compile() local
291 int input_num = branch_results[0].input_mapping[i] + 1; in Compile() local
497 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() local256 auto OnInputNumber = [&](int input_num) { in Compute()294 for (int input_num = 0; input_num < indices_inputs.size(); in Compute() local302 for (int input_num = first; input_num < last; ++input_num) { in Compute() local309 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
780 for (auto input_num : op.inputs) { in ConvertOp() local1276 for (auto input_num : op->inputs) { in ConvertSubgraph() local
641 for (int64_t input_num = 0; input_num < inputs.size(); ++input_num) { in BuildXlaCompilerArguments() local
2576 int64_t input_num = in BuildDynamicShapeNodes() local3565 const int64_t input_num = in BuildExecuteNodes() local3657 int64_t input_num = params_info.FirstBroadcastArgFromHost() + in BuildExecuteNodes() local