Lines Matching refs:input_num
104 int input_num = 0;
106 while (input_num < num_constant_args) {
107 signature->arg_values[input_num] = ctx->input(input_num);
108 ++input_num;
111 while (input_num < ctx->num_inputs() - variable_args.size()) {
112 signature->arg_types.emplace_back(ctx->input_dtype(input_num),
113 ctx->input(input_num).shape());
114 ++input_num;
119 TF_RET_CHECK(input_num < ctx->num_inputs());
126 ++input_num;
141 int input_num = 0;
145 while (input_num < num_constant_args) {
146 const Tensor& input = ctx->input(input_num);
148 XlaCompiler::Argument& arg = (*args)[input_num];
153 ++input_num;
161 while (input_num < num_constant_args + num_nonconst_args) {
162 const Tensor& input = ctx->input(input_num);
164 XlaCompiler::Argument& arg = (*args)[input_num];
173 ++input_num;
177 TF_RET_CHECK(input_num + num_variable_args == ctx->num_inputs());
179 const Tensor& input = ctx->input(input_num);
182 XlaCompiler::Argument& arg = (*args)[input_num];
201 ++input_num;
