| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor.h | 45 const TensorDescriptor& descriptor);
47 const TensorDescriptor& descriptor);
49 const BHWC& shape, const TensorDescriptor& descriptor);
51 const BHWDC& shape, const TensorDescriptor& descriptor);
71 TensorDescriptor GetDescriptor() const { return descriptor_; } in GetDescriptor()
95 absl::Status CreateFromDescriptor(const TensorDescriptor& desc,
115 TensorDescriptor descriptor_;
121 const TensorDescriptor& descriptor,
125 const TensorDescriptor& descriptor,
129 const TensorDescriptor& descriptor, Tensor* result);
[all …]
|
| D | inference_context.h | 168 TensorDescriptor descriptor;
188 std::vector<std::pair<ValueId, TensorDescriptor>> GetTensorDescs() const { in GetTensorDescs()
189 std::vector<std::pair<ValueId, TensorDescriptor>> result; in GetTensorDescs()
191 TensorDescriptor desc = v.second.descriptor; in GetTensorDescs()
202 void Add(const std::vector<std::pair<ValueId, TensorDescriptor>>& tensors) { in Add()
220 absl::flat_hash_map<ValueId, TensorDescriptor> const_tensors_descs_;
|
| D | tensor.cc | 34 const TensorDescriptor& descriptor, in AllocateTensorMemory()
217 const TensorDescriptor& descriptor, cl_mem memory, in CreateTensor()
240 const TensorDescriptor& descriptor, in CreateTensorShared()
259 const TensorDescriptor& descriptor) in Tensor()
267 const TensorDescriptor& descriptor) in Tensor()
275 const BHWC& shape, const TensorDescriptor& descriptor) in Tensor()
283 const BHWDC& shape, const TensorDescriptor& descriptor) in Tensor()
347 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(obj_ptr); in GetGPUResources()
601 absl::Status Tensor::CreateFromDescriptor(const TensorDescriptor& desc, in CreateFromDescriptor()
625 const TensorDescriptor& descriptor, Tensor* result) { in CreateTensor()
[all …]
|
| D | serialization.cc | 455 flatbuffers::Offset<data::TensorDescriptor> Encode( in Encode()
456 const TensorDescriptor& desc, flatbuffers::FlatBufferBuilder* builder) { in Encode()
479 void Decode(const data::TensorDescriptor* fb_desc, TensorDescriptor* desc) { in Decode()
554 TensorDescriptor desc; in Decode()
556 args->AddObject(key, absl::make_unique<TensorDescriptor>(std::move(desc))); in Decode()
593 TensorDescriptor desc; in Decode()
597 absl::make_unique<TensorDescriptor>(std::move(desc))); in Decode()
677 dynamic_cast<const TensorDescriptor*>(value.second.get()); in Encode()
730 dynamic_cast<const TensorDescriptor*>(value.second.get()); in Encode()
768 std::vector<flatbuffers::Offset<tflite::gpu::data::TensorDescriptor>> in Encode()
[all …]
|
| D | tensor_test.cc | 33 const TensorDescriptor& descriptor, in TensorGenericTest()
62 const TensorDescriptor& descriptor, in Tensor5DGenericTest()
|
| D | inference_context.cc | 63 std::vector<std::pair<ValueId, TensorDescriptor>> GetCLNodeTensors( in GetCLNodeTensors()
65 std::vector<std::pair<ValueId, TensorDescriptor>> result; in GetCLNodeTensors()
277 TensorDescriptor{data_type, TensorStorageType::SINGLE_TEXTURE_2D, in ReserveGraphTensors()
285 t->id, {shape, TensorDescriptor{data_type, storage_type, layout}}); in ReserveGraphTensors()
294 std::map<ValueId, TensorDescriptor> tensor_descriptors; in ConvertOperations()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | tensor_desc.cc | 76 TensorDescriptor::TensorDescriptor(TensorDescriptor&& desc) in TensorDescriptor() function in tflite::gpu::TensorDescriptor
83 TensorDescriptor& TensorDescriptor::operator=(TensorDescriptor&& desc) { in operator =()
95 GPUResources TensorDescriptor::GetGPUResources() const { in GetGPUResources()
166 absl::Status TensorDescriptor::PerformSelector( in PerformSelector()
225 absl::Status TensorDescriptor::PerformReadSelector( in PerformReadSelector()
263 absl::Status TensorDescriptor::GetLinkingContextFromWriteSelector( in GetLinkingContextFromWriteSelector()
286 absl::Status TensorDescriptor::PerformWriteSelector( in PerformWriteSelector()
302 absl::Status TensorDescriptor::PerformWriteLinearSelector( in PerformWriteLinearSelector()
318 absl::Status TensorDescriptor::PerformWrite2DSelector( in PerformWrite2DSelector()
333 std::string TensorDescriptor::Read( in Read()
[all …]
|
| D | tensor_desc.h | 45 struct TensorDescriptor : public GPUObjectDescriptor { struct
46 TensorDescriptor() = default;
47 TensorDescriptor(DataType dt, TensorStorageType st, Layout l) in TensorDescriptor() function
50 TensorDescriptor(const TensorDescriptor&) = default;
51 TensorDescriptor& operator=(const TensorDescriptor&) = default; argument
52 TensorDescriptor(TensorDescriptor&& desc);
53 TensorDescriptor& operator=(TensorDescriptor&& desc); argument
55 bool operator==(const TensorDescriptor& d) const {
60 bool operator!=(const TensorDescriptor& d) const { return !(*this == d); }
191 const TensorDescriptor& desc, T* dst);
[all …]
|
| D | storage_type_util.cc | 27 const TensorDescriptor& descriptor) { in CanCreateTensorWithShape()
73 const TensorDescriptor& descriptor) { in CanCreateTensorWithShape()
84 TensorDescriptor{data_type, desired, layout})) { in SelectBestStorageType()
94 TensorDescriptor{data_type, TensorStorageType::IMAGE_BUFFER, in SelectBestStorageType()
105 TensorDescriptor{data_type, TensorStorageType::TEXTURE_ARRAY, in SelectBestStorageType()
115 TensorDescriptor{data_type, TensorStorageType::TEXTURE_2D, in SelectBestStorageType()
|
| D | gpu_operation.h | 75 std::vector<TensorDescriptor> src_tensors;
76 std::vector<TensorDescriptor> dst_tensors;
127 const TensorDescriptor& desc);
133 const TensorDescriptor& desc);
|
| D | gpu_operation.cc | 159 const TensorDescriptor& desc) { in AddSrcTensor()
161 auto desc_new = absl::make_unique<TensorDescriptor>(desc); in AddSrcTensor()
180 const TensorDescriptor& desc) { in AddDstTensor()
182 auto desc_new = absl::make_unique<TensorDescriptor>(desc); in AddDstTensor()
189 absl::make_unique<TensorDescriptor>(definition_.src_tensors[0]); in AssembleCode()
197 absl::make_unique<TensorDescriptor>(definition_.dst_tensors[0]); in AssembleCode()
|
| D | storage_type_util.h | 29 const TensorDescriptor& descriptor);
32 const TensorDescriptor& descriptor);
|
| D | serialization_base.fbs | 149 table TensorDescriptor {
175 value:TensorDescriptor;
219 src_tensors:[TensorDescriptor];
220 dst_tensors:[TensorDescriptor];
|
| D | serialization_base_generated.h | 63 struct TensorDescriptor;
1384 struct TensorDescriptor FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
1428 typedef TensorDescriptor Table;
1432 fbb_.AddOffset(TensorDescriptor::VT_BASE_OBJ, base_obj); in add_base_obj()
1435 fbb_.AddElement<int8_t>(TensorDescriptor::VT_DATA_TYPE, static_cast<int8_t>(data_type), 0); in add_data_type()
1438 … fbb_.AddElement<int8_t>(TensorDescriptor::VT_STORAGE_TYPE, static_cast<int8_t>(storage_type), 0); in add_storage_type()
1441 fbb_.AddElement<int8_t>(TensorDescriptor::VT_LAYOUT, static_cast<int8_t>(layout), 0); in add_layout()
1444 fbb_.AddOffset(TensorDescriptor::VT_SHAPE, shape); in add_shape()
1447 fbb_.AddOffset(TensorDescriptor::VT_DATA, data); in add_data()
1453 flatbuffers::Offset<TensorDescriptor> Finish() { in Finish()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor.h | 40 const BHWC& shape, const TensorDescriptor& descriptor);
43 const BHWDC& shape, const TensorDescriptor& descriptor);
63 TensorDescriptor GetDescriptor() const { return descriptor_; } in GetDescriptor()
81 absl::Status CreateFromDescriptor(const TensorDescriptor& desc,
103 TensorDescriptor descriptor_;
107 const TensorDescriptor& descriptor,
111 const TensorDescriptor& descriptor,
115 const TensorDescriptor& descriptor,
119 const TensorDescriptor& descriptor,
|
| D | inference_context.h | 153 TensorDescriptor descriptor;
173 std::vector<std::pair<ValueId, TensorDescriptor>> GetTensorDescs() const { in GetTensorDescs()
174 std::vector<std::pair<ValueId, TensorDescriptor>> result; in GetTensorDescs()
176 TensorDescriptor desc = v.second.descriptor; in GetTensorDescs()
187 void Add(const std::vector<std::pair<ValueId, TensorDescriptor>>& tensors) { in Add()
213 std::map<ValueId, TensorDescriptor> const_tensors_descs_;
|
| D | metal_spatial_tensor.cc | 30 const TensorDescriptor& descriptor, in CreateTextureBuffer()
58 const TensorDescriptor& descriptor, in AllocateTensorMemory()
154 const TensorDescriptor& descriptor, in CreateTensor()
175 const TensorDescriptor& descriptor) in MetalSpatialTensor()
188 const TensorDescriptor& descriptor) in MetalSpatialTensor()
252 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(obj_ptr); in GetGPUResources()
506 const TensorDescriptor& desc, id<MTLDevice> device) { in CreateFromDescriptor()
546 const TensorDescriptor& descriptor, in CreateTensor()
553 const TensorDescriptor& descriptor, in CreateTensor()
559 const TensorDescriptor& descriptor, in CreateSharedBufferTensor()
[all …]
|
| D | metal_spatial_tensor_test.mm | 33 using tflite::gpu::TensorDescriptor;
41 const TensorDescriptor& descriptor,
72 const TensorDescriptor& descriptor,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | elementwise.cc | 211 TensorDescriptor desc{definition.GetDataType(), storage_type, Layout::HWC}; in CreateElementwiseTwoInput()
217 absl::make_unique<TensorDescriptor>(std::move(desc))); in CreateElementwiseTwoInput()
243 TensorDescriptor desc{definition.GetDataType(), storage_type, Layout::HWC}; in CreateElementwiseTwoInput()
249 absl::make_unique<TensorDescriptor>(std::move(desc))); in CreateElementwiseTwoInput()
|
| D | prelu.cc | 54 TensorDescriptor desc{definition.GetDataType(), storage_type, Layout::HWC}; in CreatePReLU()
57 "alpha", absl::make_unique<TensorDescriptor>(std::move(desc))); in CreatePReLU()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | converter.cc | 67 TensorDescriptor tensor_descriptor_;
130 absl::make_unique<TensorDescriptor>(src_tensor_descriptor_)); in Init()
138 absl::make_unique<TensorDescriptor>(dst_tensor_descriptor_)); in Init()
196 TensorDescriptor src_tensor_descriptor_;
197 TensorDescriptor dst_tensor_descriptor_;
218 absl::make_unique<TensorDescriptor>(tensor_descriptor_)); in Init()
316 absl::make_unique<TensorDescriptor>(tensor_descriptor_)); in Init()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | special_selector.cc | 34 const std::map<ValueId, TensorDescriptor>& tensor_descriptors, in TryDepthwiseConvPlus1x1Conv()
98 const std::map<ValueId, TensorDescriptor>& tensor_descriptors, in TryFCFCAdd()
188 const std::map<ValueId, TensorDescriptor>& tensor_descriptors, in GPUSubgraphFromGraph()
|
| D | operation_selector.cc | 97 TensorDescriptor td_0; in WinogradFromNode()
103 TensorDescriptor td_1; in WinogradFromNode()
210 TensorDescriptor transposed_desc = {op_def.src_tensors[1].data_type, in GPUOperationFromNode()
216 TensorDescriptor weights_desc = {op_def.src_tensors[1].data_type, in GPUOperationFromNode()
293 TensorDescriptor weights_desc = {op_def.src_tensors[1].data_type, in GPUOperationFromNode()
369 TensorDescriptor(op_def.GetDataType(), in GPUOperationFromNode()
379 TensorDescriptor(op_def.GetDataType(), TensorStorageType::BUFFER, in GPUOperationFromNode()
|
| D | special_selector.h | 36 const std::map<ValueId, TensorDescriptor>& tensor_descriptors,
|
| D | subgraph.h | 41 std::vector<std::pair<BHWC, TensorDescriptor>> new_tensors;
|