Searched defs:input_num (Results 1 – 15 of 15) 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
65 const long input_num = BASE_NUM * char_val; in TEST() local89 const long input_num = static_cast<int32_t>(char_val << (6 * cur_char)); in TEST() 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
81 size_t input_num = 0; in propWithNoDevice() local
293 leading_zeroes(uint64_t input_num) { in leading_zeroes()
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