Home
last modified time | relevance | path

Searched refs:arg_expressions (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc957 std::vector<XlaExpression>* arg_expressions, in BuildArguments() argument
960 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() local
1335 arg_shardings, &arg_expressions, &result->input_mapping, in CompileGraph()
1337 context->set_args(std::move(arg_expressions)); in CompileGraph()
Dxla_compiler.h308 std::vector<XlaExpression>* arg_expressions,