Searched refs:input_to_args (Results 1 – 2 of 2) sorted by relevance
707 std::vector<int>* input_to_args, std::vector<xla::Shape>* input_shapes, in BuildArguments() argument714 input_to_args->clear(); in BuildArguments()715 input_to_args->reserve(args.size()); in BuildArguments()737 input_to_args->push_back(i); in BuildArguments()743 input_to_args->push_back(i); in BuildArguments()755 if (input_to_args->empty()) { in BuildArguments()763 for (int i = 0; i < input_to_args->size(); i++) { in BuildArguments()764 arg_to_inputs[input_to_args->at(i)] = i; in BuildArguments()767 std::vector<xla::Shape> arg_shapes(input_to_args->size()); in BuildArguments()768 for (std::vector<int>::size_type i = 0; i < input_to_args->size(); ++i) { in BuildArguments()[all …]
436 std::vector<int>* input_to_args,