Searched refs:set_device_ordinal (Results 1 – 16 of 16) 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()
611 DefaultExecutableBuildOptions().set_device_ordinal(d), in XLA_TEST_F()612 DefaultExecutableRunOptions().set_device_ordinal(d)); in XLA_TEST_F()619 DefaultExecutableBuildOptions().set_device_ordinal(d), in XLA_TEST_F()620 DefaultExecutableRunOptions().set_device_ordinal(d)); in XLA_TEST_F()637 DefaultExecutableBuildOptions().set_device_ordinal( in XLA_TEST_F()639 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
250 updated_options.set_device_ordinal(default_device_ordinal()); in Compile()
20 ExecutableRunOptions& ExecutableRunOptions::set_device_ordinal( in set_device_ordinal() function in xla::ExecutableRunOptions
50 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()
142 run_options.set_device_ordinal(stream_exec->device_ordinal()); in CompareEqualImpl()
123 build_options.set_device_ordinal(client->default_device_ordinal()); in Compile()
152 build_options.set_device_ordinal(options.device_ordinal != -1 in BuildExecutable()
246 options.set_device_ordinal(device_ordinal); in Execute()294 options.set_device_ordinal(device_ordinal); in ExecutePerReplica()
445 run_options.set_device_ordinal(device); in GetServiceRunOptionsForDevice()
544 options.set_device_ordinal(stream->parent()->device_ordinal()); in ExecuteAndRegisterResult()