Searched refs:xla_args (Results 1 – 3 of 3) sorted by relevance
65 std::vector<XlaCompiler::Argument> xla_args; in ConvertGraphToXla() local66 TF_RETURN_IF_ERROR(CreateXlaArgs(*graph, &xla_args)); in ConvertGraphToXla()68 PopulateXlaArgs(config, &xla_args); in ConvertGraphToXla()86 options, "tfcompile", std::move(graph), xla_args, &result)); in ConvertGraphToXla()114 std::vector<bool> updated_inputs(xla_args.size()); in ConvertGraphToXla()118 int64 input_index = xla_args.size() - config.variable_size(); in ConvertGraphToXla()
30 std::vector<XlaCompiler::Argument>* xla_args);34 std::vector<XlaCompiler::Argument>* xla_args);
233 std::vector<XlaCompiler::Argument>* xla_args) { in CreateXlaArgs() argument244 xla_args->push_back(arg); in CreateXlaArgs()250 std::vector<XlaCompiler::Argument>* xla_args) { in PopulateXlaArgs() argument261 xla_args->push_back(std::move(arg)); in PopulateXlaArgs()