Searched refs:set_result_layout (Results 1 – 9 of 9) sorted by relevance
46 ExecutableBuildOptions& set_result_layout(const Shape& shape_with_layout);
50 ExecutableBuildOptions& ExecutableBuildOptions::set_result_layout( in set_result_layout() function in xla::ExecutableBuildOptions
87 build_options.set_result_layout(program_shape->result()); in RealMain()
64 build_options.set_result_layout(program_shape->result()); in RealMain()
171 DefaultExecutableBuildOptions().set_result_layout( in XLA_TEST_F()183 DefaultExecutableBuildOptions().set_result_layout( in XLA_TEST_F()266 options.set_result_layout(shape_with_layout); in XLA_TEST_F()589 DefaultExecutableBuildOptions().set_result_layout( in XLA_TEST_F()
124 build_options.set_result_layout(xla::Shape(config.program_shape().result())); in Compile()
155 build_options.set_result_layout(result.xla_output_shape); in BuildExecutable()
241 build_options.set_result_layout(statusor.ValueOrDie());
232 exec_options.set_result_layout(input_program_shape.result()); in XlaCompiledProgramShape()