| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_init.cc | 32 Status ValidatePluggableDeviceMachineManager(const string& platform_name) { in ValidatePluggableDeviceMachineManager()
36 se::Platform* PluggableDeviceMachineManager(const string& platform_name) { in PluggableDeviceMachineManager()
|
| D | pluggable_device_process_state.cc | 46 const string& device_type, const string& platform_name) { in singleton()
60 const string& device_type, const string& platform_name) in PluggableDeviceProcessState()
|
| D | pluggable_device_factory.cc | 78 Status SingleVirtualDeviceMemoryLimit(const string& platform_name, in SingleVirtualDeviceMemoryLimit()
117 const string& platform_name) in PluggableDeviceFactory()
|
| D | pluggable_device_plugin_init.cc | 50 string device_type, platform_name; in InitDeviceAndGraphModule() local
|
| D | pluggable_device.cc | 141 const std::string& device_type, const std::string& platform_name, in PluggableDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_compiler_test.cc | 33 absl::string_view platform_name() const override { in TEST() function in xla::__anonbcc7c7340111::TEST::PjRtTestTopology
51 absl::string_view platform_name() const override { return "registered"; } in TEST() function in xla::__anonbcc7c7340111::TEST::PjRtTestTopology
|
| D | pjrt_compiler.cc | 36 void PjRtRegisterCompiler(absl::string_view platform_name, in PjRtRegisterCompiler()
|
| D | gpu_device.cc | 154 const std::optional<std::string>& platform_name, in GetGpuXlaClient()
426 std::optional<std::string> platform_name) { in GetGpuClient()
|
| D | pjrt_stream_executor_client.h | 181 absl::string_view platform_name() const override { return platform_name_; } in platform_name() function
570 absl::string_view platform_name() const { return client_->platform_name(); } in platform_name() function
|
| D | pjrt_c_api_client.cc | 122 absl::string_view PjRtCApiClient::platform_name() const { in platform_name() function in xla::PjRtCApiClient
129 absl::string_view platform_name(args.platform_name, args.platform_name_size); in platform_name() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 41 std::string platform_name, GpuDeviceInfo gpu_device_info, in IrEmitterContext()
62 absl::string_view platform_name() const { return platform_name_; } in platform_name() function
|
| D | ir_emitter_unnested.h | 123 absl::string_view platform_name() const { in platform_name() function
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_device.cc | 39 ResourceMgr* GetResourceMgr(const std::string& platform_name) { in GetResourceMgr()
104 xla::LocalClient* client, int ordinal, const std::string& platform_name, in Acquire()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | platform_util.cc | 45 std::string CanonicalPlatformName(const std::string& platform_name) { in CanonicalPlatformName()
117 const std::string& platform_name) { in GetPlatform()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| D | lmhlo_to_gpu_binary.h | 51 std::string platform_name; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/translate/ |
| D | convert_xla_gpu.cc | 35 llvm::StringRef platform_name) { in ConvertXlaGpuToGpuProgram()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | hlo_to_llvm_ir.cc | 73 std::string platform_name = "CUDA"; in CompileAndPrintLlvmIr() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_client.h | 100 absl::string_view platform_name() const { in platform_name() function
|
| D | py_buffer.h | 76 absl::string_view platform_name() const { in platform_name() function
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.h | 144 const absl::string_view platform_name() const { return platform_name_; } in platform_name() function
237 const absl::string_view platform_name() const { in platform_name() function
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | plugin_registry.cc | 113 std::string platform_name = "<unregistered platform>"; in SetDefaultFactory() local
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 40 std::string device_type, platform_name; in TEST() local
64 std::string device_type, platform_name; in TEST() local
79 std::string device_type, platform_name; in TEST() local
96 std::string device_type, platform_name; in TEST() local
112 std::string device_type, platform_name; in TEST() local
128 std::string device_type, platform_name; in TEST() local
|
| /external/OpenCL-CTS/test_conformance/vulkan/ |
| D | main.cpp | 127 char platform_name[1024] = ""; variable
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 94 explicit LLVMCompilerTest(std::string platform_name) in LLVMCompilerTest()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | while_op_test.cc | 100 const std::string platform_name = "MY_TEST"; in TEST_F() local
|