Searched refs:tuple_arguments (Results 1 – 4 of 4) sorted by relevance
523 xla::Shape result_shape, bool tuple_arguments) in PyTpuExecutable() argument526 tuple_arguments_(tuple_arguments), in PyTpuExecutable()808 std::shared_ptr<PyTpuClient> client, bool tuple_arguments) { in Compile() argument874 std::move(client), std::move(result_layout), tuple_arguments); in Compile()882 bool tuple_arguments) { in CompileMlir() argument885 /*use_tuple_args=*/tuple_arguments, in CompileMlir()888 tuple_arguments); in CompileMlir()
339 std::shared_ptr<PyTpuClient> client, bool tuple_arguments);344 std::shared_ptr<PyTpuClient> client, bool tuple_arguments);349 xla::Shape result_shape, bool tuple_arguments);
41 std::vector<HloInstruction*> tuple_arguments(shape.tuple_shapes_size()); in CreateConstant() local43 tuple_arguments[index] = in CreateConstant()47 HloInstruction::CreateTuple(tuple_arguments)); in CreateConstant()
212 tuple_arguments: bool