Searched refs:metal_device (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | BUILD | 115 ":metal_device",
152 ":metal_device",
207 ":metal_device",
220 name = "metal_device",
221 srcs = ["metal_device.cc"],
222 hdrs = ["metal_device.h"],
|
| D | inference_context.cc | 151 MetalDevice metal_device(device_id); in InitFromGraph() local
152 ReserveGraphTensors(create_info, metal_device.GetInfo(), graph, in InitFromGraph()
154 RETURN_IF_ERROR(Compile(graph, metal_device.GetInfo(), create_info.hints)); in InitFromGraph()
156 RETURN_IF_ERROR(CompileOperations(&metal_device)); in InitFromGraph()
157 RETURN_IF_ERROR(AllocateTensors(&metal_device, preallocated_ids)); in InitFromGraph()
159 RETURN_IF_ERROR(UpdateParams(metal_device.GetInfo())); in InitFromGraph()
160 RETURN_IF_ERROR(Tune(TuningType::kFast, &metal_device)); in InitFromGraph()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | BUILD | 584 "//tensorflow/lite/delegates/gpu/metal:metal_device",
|