Searched refs:set_device_ordinal (Results 1 – 14 of 14) sorted by relevance
92 computation, {}, ExecutableBuildOptions().set_device_ordinal(d), in XLA_TEST_F()93 ExecutableRunOptions().set_device_ordinal(d).set_allocator(allocator)); in XLA_TEST_F()
614 DefaultExecutableBuildOptions().set_device_ordinal(d), in XLA_TEST_F()615 DefaultExecutableRunOptions().set_device_ordinal(d)); in XLA_TEST_F()622 DefaultExecutableBuildOptions().set_device_ordinal(d), in XLA_TEST_F()623 DefaultExecutableRunOptions().set_device_ordinal(d)); in XLA_TEST_F()640 DefaultExecutableBuildOptions().set_device_ordinal( in XLA_TEST_F()642 DefaultExecutableRunOptions().set_device_ordinal( in XLA_TEST_F()
43 execute_options.set_device_ordinal(device_ordinal); in CompileAndExecute()
39 ExecutableBuildOptions& set_device_ordinal(int device_ordinal);
34 ExecutableBuildOptions& ExecutableBuildOptions::set_device_ordinal( in set_device_ordinal() function in xla::ExecutableBuildOptions
182 options.set_device_ordinal(-1); in Run()346 updated_options.set_device_ordinal(default_device_ordinal()); in Compile()
33 ExecutableRunOptions& ExecutableRunOptions::set_device_ordinal( in set_device_ordinal() function in xla::ExecutableRunOptions
80 ExecutableRunOptions& set_device_ordinal(int device_ordinal);
86 build_options.set_device_ordinal(0); in RealMain()
63 build_options.set_device_ordinal(0); in RealMain()
122 build_options.set_device_ordinal(client->default_device_ordinal()); in Compile()
159 build_options.set_device_ordinal(options.device_ordinal != -1 in BuildExecutable()
450 run_options.set_device_ordinal(device); in GetServiceRunOptionsForDevice()
562 options.set_device_ordinal(stream->parent()->device_ordinal()); in ExecuteAndRegisterResult()