Searched refs:arg_handles (Results 1 – 1 of 1) sorted by relevance
788 std::vector<xla::XlaOp> arg_handles(input_to_args->size()); in BuildArguments() local824 arg_handles[i] = xla::GetTupleElement(tuple, i); in BuildArguments()833 arg_handles[i] = xla::Parameter(builder, i, (*input_shapes)[i], in BuildArguments()865 arg_handles[i], builder)); in BuildArguments()876 xla::Reshape(arg_handles[i], arg.DimensionSizes()), arg.type); in BuildArguments()878 arg_expression = XlaExpression::XlaOp(arg_handles[i], arg.type); in BuildArguments()882 arg_expression = XlaExpression::XlaOp(arg_handles[i], arg.type); in BuildArguments()