Searched refs:set_num_partitions (Results 1 – 13 of 13) sorted by relevance
68 ExecutableBuildOptions& ExecutableBuildOptions::set_num_partitions( in set_num_partitions() function in xla::ExecutableBuildOptions126 execution_options.set_num_partitions(build_options.num_partitions()); in CreateExecutionOptions()
85 ExecutableBuildOptions& set_num_partitions(int num_partitions);
84 execution_options.set_num_partitions(options.num_cores()); in CompileAheadOfTime()
94 config->set_num_partitions(execution_options->num_partitions()); in CreateModuleConfig()
131 void set_num_partitions(int64 num_partitions) { in set_num_partitions() function
638 config.set_num_partitions(2); in TEST_F()
445 module_config.set_num_partitions(execution_options->num_partitions()); in CreateModuleConfigFromShape()
472 options.executable_build_options.set_num_partitions(num_partitions); in BuildXlaCompilerSubmodule()540 &ExecutableBuildOptions::set_num_partitions) in BuildXlaCompilerSubmodule()
401 compile_options.executable_build_options.set_num_partitions(1); in SendShutdownOutfeedHeader()
40 compile_options.executable_build_options.set_num_partitions(1); in CompileAndExecute()
100 config->set_num_partitions(num_partitions); in CreateModuleConfig()
460 config.set_num_partitions(c_config.num_partitions); in FromC()
801 build_options.set_num_partitions(2); in XLA_TEST_F()