Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/
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;
Dapi.h215 struct TensorObjectDef { struct
220 bool operator==(const TensorObjectDef& other) const { argument
226 bool IsValid(const TensorObjectDef& def);
229 uint32_t NumElements(const TensorObjectDef& def);
236 bool IsValid(const TensorObjectDef& def, const TensorObject& object);
256 virtual std::vector<TensorObjectDef> inputs() const = 0;
257 virtual std::vector<TensorObjectDef> outputs() const = 0;
303 virtual std::vector<TensorObjectDef> inputs() const = 0;
304 virtual std::vector<TensorObjectDef> outputs() const = 0;
Dapi.cc73 bool IsValid(const TensorObjectDef& def) { return IsValid(def.object_def); } in IsValid()
75 bool IsValid(const TensorObjectDef& def, const TensorObject& object) { in IsValid()
105 uint32_t NumElements(const TensorObjectDef& def) { in NumElements()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc41 virtual absl::Status Init(const TensorObjectDef& input_def,
42 const TensorObjectDef& output_def,
120 absl::Status Init(const TensorObjectDef& input_def, in Init()
121 const TensorObjectDef& output_def, in Init()
208 absl::Status Init(const TensorObjectDef& input_def, in Init()
209 const TensorObjectDef& output_def, in Init()
291 const TensorObjectDef& input_def, in GetFromBhwcKernel()
292 const TensorObjectDef& output_def) const { in GetFromBhwcKernel()
304 absl::Status Init(const TensorObjectDef& input_def, in Init()
305 const TensorObjectDef& output_def, in Init()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconverter.cc52 virtual absl::Status Init(const TensorObjectDef& input_def,
53 const TensorObjectDef& output_def) = 0;
106 absl::Status Init(const TensorObjectDef& input_def, in Init()
107 const TensorObjectDef& output_def) final { in Init()
194 absl::Status Init(const TensorObjectDef& input_def, in Init()
195 const TensorObjectDef& output_def) final { in Init()
347 bool IsSupported(const TensorObjectDef& input, in IsSupported()
348 const TensorObjectDef& output) const final { in IsSupported()
359 const TensorObjectDef& input, const TensorObjectDef& output, in MakeConverter()
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()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dapi2.cc60 absl::Status MaybeAllocateGlBuffer(const TensorObjectDef& def, GlBuffer* ssbo) { in MaybeAllocateGlBuffer()
224 const TensorObjectDef& d = def().internal_def; in MaybeAllocateInternalObject()
244 const TensorObjectDef& d = def().external_def; in MaybeAllocateExternalObject()
410 std::vector<TensorObjectDef> inputs() const override { in inputs()
414 std::vector<TensorObjectDef> outputs() const override { in outputs()
477 static std::vector<TensorObjectDef> GetExternalDefinitions( in GetExternalDefinitions()
479 std::vector<TensorObjectDef> defs; in GetExternalDefinitions()
511 std::vector<TensorObjectDef> inputs() const final { in inputs()
515 std::vector<TensorObjectDef> outputs() const final { in outputs()
616 TensorObjectDef external_def; in LinkTensors()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dapi.cc179 const TensorObjectDef& d = def().external_def; in MaybeAllocateExternalObject()
468 std::vector<TensorObjectDef> inputs() const override { in inputs()
472 std::vector<TensorObjectDef> outputs() const override { in outputs()
565 static std::vector<TensorObjectDef> GetExternalDefinitions( in GetExternalDefinitions()
567 std::vector<TensorObjectDef> defs; in GetExternalDefinitions()
584 TensorObjectDef TensorToDef(const Tensor& tensor) { in TensorToDef()
585 TensorObjectDef def; in TensorToDef()
718 std::vector<TensorObjectDef> inputs() const override { in inputs()
722 std::vector<TensorObjectDef> outputs() const override { in outputs()
787 TensorObjectDef def = TensorToDef(*context_->GetTensor(id)); in LinkTensors()
[all …]
Dgl_interop.h155 GlClBufferCopier(const TensorObjectDef& input_def,
156 const TensorObjectDef& output_def, Environment* environment);
Dgl_interop.cc263 GlClBufferCopier::GlClBufferCopier(const TensorObjectDef& input_def, in GlClBufferCopier()
264 const TensorObjectDef& output_def, in GlClBufferCopier()
Dgpu_api_delegate.cc240 TensorObjectDef def; in BindGlBufferToTensor()
301 std::vector<std::pair<TensorObject, TensorObjectDef>> tensors_;