Searched refs:input_to_args (Results 1 – 2 of 2) sorted by relevance
958 std::vector<int>* input_to_args, std::vector<xla::Shape>* input_shapes, in BuildArguments() argument965 input_to_args->clear(); in BuildArguments()966 input_to_args->reserve(args.size()); in BuildArguments()992 input_to_args->push_back(i); in BuildArguments()999 input_to_args->push_back(i); in BuildArguments()1011 if (input_to_args->empty() && !use_tuple_arg) { in BuildArguments()1019 for (int i = 0, end = input_to_args->size(); i < end; i++) { in BuildArguments()1020 arg_to_inputs[input_to_args->at(i)] = i; in BuildArguments()1023 std::vector<xla::Shape> arg_shapes(input_to_args->size()); in BuildArguments()1024 for (std::vector<int>::size_type i = 0; i < input_to_args->size(); ++i) { in BuildArguments()[all …]
309 std::vector<int>* input_to_args,