Searched refs:arg_expressions (Results 1 – 2 of 2) sorted by relevance
957 std::vector<XlaExpression>* arg_expressions, in BuildArguments() argument960 arg_expressions->resize(args.size()); in BuildArguments()972 XlaExpression& arg_expression = (*arg_expressions)[i]; in BuildArguments()1141 XlaExpression& arg_expression = (*arg_expressions)[input_to_args->at(i)]; in BuildArguments()1332 std::vector<XlaExpression> arg_expressions; in CompileGraph() local1335 arg_shardings, &arg_expressions, &result->input_mapping, in CompileGraph()1337 context->set_args(std::move(arg_expressions)); in CompileGraph()
308 std::vector<XlaExpression>* arg_expressions,