Home
last modified time | relevance | path

Searched refs:MetalDevice (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_device.h31 class MetalDevice {
33 MetalDevice();
34 MetalDevice(id<MTLDevice> device);
36 MetalDevice(MetalDevice&& device) = default;
37 MetalDevice& operator=(MetalDevice&& device) = default;
38 MetalDevice(const MetalDevice&) = delete;
39 MetalDevice& operator=(const MetalDevice&) = delete;
41 ~MetalDevice() = default;
Dmetal_device.cc67 MetalDevice::MetalDevice() : device_(MTLCreateSystemDefaultDevice()) { in MetalDevice() function in tflite::gpu::metal::MetalDevice
70 MetalDevice::MetalDevice(id<MTLDevice> device) : device_(device) { in MetalDevice() function in tflite::gpu::metal::MetalDevice
74 bool MetalDevice::IsLanguageVersion2orHigher() const { in IsLanguageVersion2orHigher()
Dcompute_task.h57 absl::Status Compile(MetalDevice* device);
68 absl::Status Tune(TuningType tuning_type, MetalDevice* device);
71 absl::Status CompileProgram(MetalDevice* device,
Dinference_context.h135 absl::Status CompileOperations(MetalDevice* device);
138 absl::Status AllocateTensors(MetalDevice* device,
140 absl::Status AllocateMemoryForConstTensors(MetalDevice* device);
141 absl::Status AllocateMemoryForBuffers(MetalDevice* device);
142 absl::Status AllocateMemoryForStrongShapes(MetalDevice* device);
149 absl::Status Tune(TuningType tuning_type, MetalDevice* device);
Dcompute_task.cc79 absl::Status ComputeTask::Compile(MetalDevice* device) { in Compile()
92 absl::Status ComputeTask::CompileProgram(MetalDevice* device, in CompileProgram()
242 absl::Status ComputeTask::Tune(TuningType tuning_type, MetalDevice* device) { in Tune()
Dinference_context.cc151 MetalDevice metal_device(device_id); in InitFromGraph()
355 absl::Status InferenceContext::CompileOperations(MetalDevice* device) { in CompileOperations()
363 MetalDevice* device, const std::set<ValueId>& preallocated_ids) { in AllocateTensors()
469 MetalDevice* device) { in AllocateMemoryForConstTensors()
478 absl::Status InferenceContext::AllocateMemoryForBuffers(MetalDevice* device) { in AllocateMemoryForBuffers()
545 MetalDevice* device) { in AllocateMemoryForStrongShapes()
588 MetalDevice* device) { in Tune()
Dmetal_arguments.h39 MetalDevice* device, Arguments* args, std::string* code);
Dmetal_arguments.cc180 const std::map<std::string, std::string>& linkables, MetalDevice* device, in Init()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtest_util.h54 MetalDevice device_;