Searched defs:input_num (Results 1 – 12 of 12) 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
278 int input_num = branch_results[0].input_mapping[i] + 1; in Compile() local
299 int input_num = then_result.input_mapping[i] + 1; in Compile() local
500 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
62 const int64 input_num = in SparseConcatKernel() local101 const int64 input_num = in SparseConcatPermuteKernel() local
54 for (int input_num = 0; input_num < TF_NumInputs(ctx); ++input_num) { in MergeSummaryOp_Compute() local
62 int input_num = -1; in ApplyToNode() local
596 for (int input_num = 0; input_num < new_node_inputs.size(); ++input_num) { in Run() local
2524 int64_t input_num = in BuildDynamicShapeNodes() local3514 const int64_t input_num = in BuildExecuteNodes() local3606 int64_t input_num = params_info.FirstBroadcastArgFromHost() + in BuildExecuteNodes() local
809 for (auto input_num : op.inputs) { in ConvertOp() local1300 for (auto input_num : op->inputs) { in ConvertSubgraph() local
645 for (int64_t input_num = 0; input_num < inputs.size(); ++input_num) { in BuildXlaCompilerArguments() local