Home
last modified time | relevance | path

Searched refs:TensorObject (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/
Dspi.h36 virtual absl::Status Convert(const TensorObject& input,
37 const TensorObject& output) = 0;
69 virtual absl::Status SetExternalObject(TensorObject obj) = 0;
71 virtual TensorObject GetExternalObject() = 0;
Dapi.h231 using TensorObject = variable
236 bool IsValid(const TensorObjectDef& def, const TensorObject& object);
238 ObjectType GetType(const TensorObject& object);
241 bool IsObjectPresent(ObjectType type, const TensorObject& obj);
245 bool IsObjectInitialized(const TensorObject& obj);
310 virtual absl::Status GetInputObject(int index, TensorObject* object) = 0;
311 virtual absl::Status GetOutputObject(int index, TensorObject* object) = 0;
312 virtual absl::Status SetInputObject(int index, TensorObject object) = 0;
313 virtual absl::Status SetOutputObject(int index, TensorObject object) = 0;
Dapi.cc69 ObjectType GetType(const TensorObject& object) { in GetType()
75 bool IsValid(const TensorObjectDef& def, const TensorObject& object) { in IsValid()
80 bool IsObjectPresent(ObjectType type, const TensorObject& obj) { in IsObjectPresent()
101 bool IsObjectInitialized(const TensorObject& obj) { in IsObjectInitialized()
Ddelegate.cc251 TensorObject GetTensorObject(int index, TfLiteContext* context) const { in GetTensorObject()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dapi.cc58 NoopTensorTie(const TensorTieDef& def, TensorObject obj) in NoopTensorTie()
65 absl::Status SetExternalObject(TensorObject obj) final { in SetExternalObject()
76 TensorObject GetExternalObject() final { return obj_; } in GetExternalObject()
83 TensorObject obj_;
90 DefaultTensorTie(const TensorTieDef& def, TensorObject internal_obj) in DefaultTensorTie()
111 static absl::Status New(const TensorTieDef& def, TensorObject internal_object, in New()
134 absl::Status SetExternalObject(TensorObject obj) final { in SetExternalObject()
145 TensorObject GetExternalObject() final { return external_obj_; } in GetExternalObject()
214 const TensorObject internal_obj_;
215 TensorObject external_obj_;
[all …]
Dgl_interop.h158 absl::Status Convert(const TensorObject& input_obj,
159 const TensorObject& output_obj) override;
Dgpu_api_delegate.cc248 TensorObject obj = buffer; in BindGlBufferToTensor()
264 TensorObject GetTensorObject(int index, TfLiteContext* context) const { in GetTensorObject()
301 std::vector<std::pair<TensorObject, TensorObjectDef>> tensors_;
Dgl_interop.cc271 absl::Status GlClBufferCopier::Convert(const TensorObject& input_obj, in Convert()
272 const TensorObject& output_obj) { in Convert()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconverter.cc138 absl::Status Convert(const TensorObject& input_obj, in Convert()
139 const TensorObject& output_obj) override { in Convert()
233 absl::Status Convert(const TensorObject& input_obj, in Convert()
234 const TensorObject& output_obj) override { in Convert()
282 absl::Status Convert(const TensorObject& input_obj, in Convert()
283 const TensorObject& output_obj) override { in Convert()
316 absl::Status Convert(const TensorObject& input_obj, in Convert()
317 const TensorObject& output_obj) override { in Convert()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc100 absl::Status GetOpenCLMemory(const TensorObject& obj, cl_mem* memory) { in GetOpenCLMemory()
172 absl::Status Convert(const TensorObject& input_obj, in Convert()
173 const TensorObject& output_obj) override { in Convert()
265 absl::Status Convert(const TensorObject& input_obj, in Convert()
266 const TensorObject& output_obj) override { in Convert()
360 absl::Status Convert(const TensorObject& input_obj, in Convert()
361 const TensorObject& output_obj) override { in Convert()
426 absl::Status Convert(const TensorObject& input_obj, in Convert()
427 const TensorObject& output_obj) override { in Convert()
489 absl::Status Convert(const TensorObject& input_obj, in Convert()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dapi2.cc81 DefaultTensorTie(const TensorTieDef& def, TensorObject internal_obj, in DefaultTensorTie()
97 absl::make_unique<DefaultTensorTie>(def, TensorObject{}, objects); in New()
105 TensorObject internal_object, in New()
132 absl::Status SetExternalObject(TensorObject obj) final { in SetExternalObject()
160 TensorObject GetExternalObject() final { return external_obj_; } in GetExternalObject()
268 TensorObject internal_obj_;
269 TensorObject external_obj_;
314 absl::Status SetExternalObject(TensorObject obj) final { in SetExternalObject()
318 TensorObject GetExternalObject() final { in GetExternalObject()
418 absl::Status GetInputObject(int index, TensorObject* object) override { in GetInputObject()
[all …]