Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
H A Dconverter.cc45 virtual absl::Status Init(const TensorObjectDef& input_def,
46 const TensorObjectDef& output_def,
125 absl::Status Init(const TensorObjectDef& input_def, in Init()
126 const TensorObjectDef& output_def, in Init()
221 absl::Status Init(const TensorObjectDef& input_def, in Init()
222 const TensorObjectDef& output_def, in Init()
314 const TensorObjectDef& input_def, in GetFromBhwcKernel()
315 const TensorObjectDef& output_def) const { in GetFromBhwcKernel()
327 absl::Status Init(const TensorObjectDef& input_def, in Init()
328 const TensorObjectDef& output_def, in Init()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/
H A Dspi.h44 virtual bool IsSupported(const TensorObjectDef& input,
45 const TensorObjectDef& output) const = 0;
48 const TensorObjectDef& input, const TensorObjectDef& output,
57 TensorObjectDef internal_def;
58 TensorObjectDef external_def;
H A Dapi.h216 struct TensorObjectDef { struct
221 bool operator==(const TensorObjectDef& other) const { argument
227 bool IsValid(const TensorObjectDef& def);
230 uint32_t NumElements(const TensorObjectDef& def);
237 bool IsValid(const TensorObjectDef& def, const TensorObject& object);
257 virtual std::vector<TensorObjectDef> inputs() const = 0;
258 virtual std::vector<TensorObjectDef> outputs() const = 0;
304 virtual std::vector<TensorObjectDef> inputs() const = 0;
305 virtual std::vector<TensorObjectDef> outputs() const = 0;
H A Dapi.cc75 bool IsValid(const TensorObjectDef& def) { return IsValid(def.object_def); } in IsValid()
77 bool IsValid(const TensorObjectDef& def, const TensorObject& object) { in IsValid()
107 uint32_t NumElements(const TensorObjectDef& def) { in NumElements()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
H A Dconverter.cc57 virtual absl::Status Init(const TensorObjectDef& input_def,
58 const TensorObjectDef& output_def) = 0;
111 absl::Status Init(const TensorObjectDef& input_def, in Init()
112 const TensorObjectDef& output_def) final { in Init()
199 absl::Status Init(const TensorObjectDef& input_def, in Init()
200 const TensorObjectDef& output_def) final { in Init()
352 bool IsSupported(const TensorObjectDef& input, in IsSupported()
353 const TensorObjectDef& output) const final { in IsSupported()
364 const TensorObjectDef& input, const TensorObjectDef& output, in MakeConverter()
H A Dconverter_test.cc71 TensorObjectDef input_def; in RunFromTensorTest()
76 TensorObjectDef output_def = input_def; in RunFromTensorTest()
128 TensorObjectDef input_def; in RunToTensorTest()
133 TensorObjectDef output_def = input_def; in RunToTensorTest()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
H A Dapi2.cc63 absl::Status MaybeAllocateGlBuffer(const TensorObjectDef& def, GlBuffer* ssbo) { in MaybeAllocateGlBuffer()
227 const TensorObjectDef& d = def().internal_def; in MaybeAllocateInternalObject()
247 const TensorObjectDef& d = def().external_def; in MaybeAllocateExternalObject()
413 std::vector<TensorObjectDef> inputs() const override { in inputs()
417 std::vector<TensorObjectDef> outputs() const override { in outputs()
480 static std::vector<TensorObjectDef> GetExternalDefinitions( in GetExternalDefinitions()
482 std::vector<TensorObjectDef> defs; in GetExternalDefinitions()
514 std::vector<TensorObjectDef> inputs() const final { in inputs()
518 std::vector<TensorObjectDef> outputs() const final { in outputs()
619 TensorObjectDef external_def; in LinkTensors()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dapi.cc184 const TensorObjectDef& d = def().external_def; in MaybeAllocateExternalObject()
472 std::vector<TensorObjectDef> inputs() const override { in inputs()
476 std::vector<TensorObjectDef> outputs() const override { in outputs()
577 static std::vector<TensorObjectDef> GetExternalDefinitions( in GetExternalDefinitions()
579 std::vector<TensorObjectDef> defs; in GetExternalDefinitions()
596 TensorObjectDef TensorToDef(const Tensor& tensor) { in TensorToDef()
597 TensorObjectDef def; in TensorToDef()
733 std::vector<TensorObjectDef> inputs() const override { in inputs()
737 std::vector<TensorObjectDef> outputs() const override { in outputs()
802 TensorObjectDef def = TensorToDef(*context_->GetTensor(id)); in LinkTensors()
[all …]
H A Dgl_interop.h155 GlClBufferCopier(const TensorObjectDef& input_def,
156 const TensorObjectDef& output_def, Environment* environment);
H A Dgl_interop.cc266 GlClBufferCopier::GlClBufferCopier(const TensorObjectDef& input_def, in GlClBufferCopier()
267 const TensorObjectDef& output_def, in GlClBufferCopier()
H A Dgpu_api_delegate.cc239 TensorObjectDef def; in BindGlBufferToTensor()
300 std::vector<std::pair<TensorObject, TensorObjectDef>> tensors_;