Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A DBUILD113 ":metal_spatial_tensor",
152 ":metal_spatial_tensor",
200 ":metal_spatial_tensor",
222 name = "metal_spatial_tensor",
223 srcs = ["metal_spatial_tensor.cc"],
224 hdrs = ["metal_spatial_tensor.h"],
249 ":metal_spatial_tensor",
294 "//tensorflow/lite/delegates/gpu/metal:metal_spatial_tensor",
H A Dcompute_task.cc253 const auto* metal_spatial_tensor = in UpdateParams() local
255 if (!metal_spatial_tensor) { in UpdateParams()
259 operation_->GetSrcTensorsNames()[i], *metal_spatial_tensor)); in UpdateParams()
262 const auto* metal_spatial_tensor = in UpdateParams() local
264 if (!metal_spatial_tensor) { in UpdateParams()
268 operation_->GetDstTensorsNames()[i], *metal_spatial_tensor)); in UpdateParams()
H A Dinference_context.cc156 auto* metal_spatial_tensor = in InitFromGraph() local
158 if (!metal_spatial_tensor) { in InitFromGraph()
161 external_immutable_tensors_[external_tensor.first] = metal_spatial_tensor; in InitFromGraph()
H A Dmetal_spatial_tensor_test.mm16 #include "tensorflow/lite/delegates/gpu/metal/metal_spatial_tensor.h"
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
H A DBUILD106 "//tensorflow/lite/delegates/gpu/metal:metal_spatial_tensor",
784 "//tensorflow/lite/delegates/gpu/metal:metal_spatial_tensor",
903 "//tensorflow/lite/delegates/gpu/metal:metal_spatial_tensor",
H A Dconv_test.mm36 #include "tensorflow/lite/delegates/gpu/metal/metal_spatial_tensor.h"
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/
H A DBUILD97 "//tensorflow/lite/delegates/gpu/metal:metal_spatial_tensor",
H A Dmetal_delegate.mm48 #include "tensorflow/lite/delegates/gpu/metal/metal_spatial_tensor.h"
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A DCMakeLists.txt338 metal_spatial_tensor