| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | tensor_linear_desc.h | 32 struct TensorLinearDescriptor : public GPUObjectDescriptor { struct
41 TensorLinearDescriptor() = default; argument
42 TensorLinearDescriptor(const TensorLinearDescriptor&) = default;
43 TensorLinearDescriptor& operator=(const TensorLinearDescriptor&) = default; argument
44 TensorLinearDescriptor(TensorLinearDescriptor&& desc) = default;
45 TensorLinearDescriptor& operator=(TensorLinearDescriptor&& desc) = default; argument
|
| D | tensor_linear_desc.cc | 27 void TensorLinearDescriptor::Release() { data.clear(); } in Release()
29 GPUResources TensorLinearDescriptor::GetGPUResources() const { in GetGPUResources()
48 absl::Status TensorLinearDescriptor::PerformSelector( in PerformSelector()
70 absl::Status TensorLinearDescriptor::PerformReadSelector( in PerformReadSelector()
98 void TensorLinearDescriptor::UploadLinearData( in UploadLinearData()
|
| D | serialization_base.fbs | 114 table TensorLinearDescriptor {
170 value:TensorLinearDescriptor;
|
| D | serialization_base_generated.h | 57 struct TensorLinearDescriptor;
1191 struct TensorLinearDescriptor FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
1234 typedef TensorLinearDescriptor Table;
1238 fbb_.AddOffset(TensorLinearDescriptor::VT_BASE_OBJ, base_obj); in add_base_obj()
1241 …fbb_.AddElement<int8_t>(TensorLinearDescriptor::VT_STORAGE_TYPE, static_cast<int8_t>(storage_type)… in add_storage_type()
1244 …fbb_.AddElement<int8_t>(TensorLinearDescriptor::VT_ELEMENT_TYPE, static_cast<int8_t>(element_type)… in add_element_type()
1247 …fbb_.AddElement<int8_t>(TensorLinearDescriptor::VT_MEMORY_TYPE, static_cast<int8_t>(memory_type), … in add_memory_type()
1250 fbb_.AddElement<int32_t>(TensorLinearDescriptor::VT_SIZE, size, 0); in add_size()
1253 fbb_.AddOffset(TensorLinearDescriptor::VT_DATA, data); in add_data()
1259 flatbuffers::Offset<TensorLinearDescriptor> Finish() { in Finish()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.cc | 182 TensorLinearDescriptor desc0; in CreateFCFCAdd()
187 "biases0", absl::make_unique<TensorLinearDescriptor>(std::move(desc0))); in CreateFCFCAdd()
189 TensorLinearDescriptor desc1; in CreateFCFCAdd()
194 "biases1", absl::make_unique<TensorLinearDescriptor>(std::move(desc1))); in CreateFCFCAdd()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | depthwise_conv.cc | 262 TensorLinearDescriptor desc; in CreateDepthwiseConvolution2D()
268 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateDepthwiseConvolution2D()
288 TensorLinearDescriptor desc; in CreateDepthwiseConvolution2DDynamicWeights()
296 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateDepthwiseConvolution2DDynamicWeights()
330 TensorLinearDescriptor desc; in CreateDepthwiseConvolution3D()
336 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateDepthwiseConvolution3D()
|
| D | winograd.cc | 431 TensorLinearDescriptor desc; in UploadBt()
436 absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in UploadBt()
507 TensorLinearDescriptor bias_desc; in CreateWinograd36To4x4()
511 desc.args_.AddObject("biases", absl::make_unique<TensorLinearDescriptor>( in CreateWinograd36To4x4()
650 TensorLinearDescriptor desc; in UploadAt()
655 absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in UploadAt()
704 TensorLinearDescriptor desc; in CreateWinograd36To4x4Tile4x1()
709 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateWinograd36To4x4Tile4x1()
|
| D | convolution_transposed_3x3_thin.cc | 241 TensorLinearDescriptor desc; in CreateConvolutionTransposed3x3Thin()
246 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateConvolutionTransposed3x3Thin()
263 TensorLinearDescriptor desc; in CreateConvolutionTransposed3x3ThinDynamicWeights()
268 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateConvolutionTransposed3x3ThinDynamicWeights()
|
| D | prelu.cc | 36 TensorLinearDescriptor desc; in CreatePReLU()
42 "alpha", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreatePReLU()
|
| D | fully_connected.cc | 163 TensorLinearDescriptor desc; in CreateFullyConnected()
173 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateFullyConnected()
|
| D | convolution_transposed_4x4.cc | 432 TensorLinearDescriptor desc; in CreateConvolutionTransposed4x4()
439 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateConvolutionTransposed4x4()
457 TensorLinearDescriptor desc; in CreateConvolutionTransposed4x4DynamicWeights()
464 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateConvolutionTransposed4x4DynamicWeights()
|
| D | convolution_transposed_3x3.cc | 428 TensorLinearDescriptor desc; in CreateConvolutionTransposed3x3()
433 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateConvolutionTransposed3x3()
452 TensorLinearDescriptor desc; in CreateConvolutionTransposed3x3DynamicWeights()
457 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateConvolutionTransposed3x3DynamicWeights()
|
| D | convolution_transposed.cc | 585 TensorLinearDescriptor desc; in CreateConvolutionTransposed()
591 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateConvolutionTransposed()
602 TensorLinearDescriptor desc; in CreateConvolutionTransposed3D()
608 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateConvolutionTransposed3D()
638 TensorLinearDescriptor desc; in CreateConvolutionTransposedDynamicWeights()
643 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateConvolutionTransposedDynamicWeights()
|
| D | conv_buffer_1x1.h | 160 TensorLinearDescriptor desc; in UploadBiases()
166 absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in UploadBiases()
|
| D | conv_constants.cc | 297 TensorLinearDescriptor desc; in CreateConvConstants()
303 "biases", absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in CreateConvConstants()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | linear_storage.cc | 56 dynamic_cast<const TensorLinearDescriptor*>(obj_ptr); in GetGPUResources()
74 const TensorLinearDescriptor& desc, CLContext* context) { in CreateFromTensorLinearDescriptor()
|
| D | linear_storage.h | 55 const TensorLinearDescriptor& desc, CLContext* context);
|
| D | serialization.cc | 426 flatbuffers::Offset<data::TensorLinearDescriptor> Encode( in Encode()
427 const TensorLinearDescriptor& desc, in Encode()
443 void Decode(const data::TensorLinearDescriptor* fb_desc, in Decode()
444 TensorLinearDescriptor* desc) { in Decode()
545 TensorLinearDescriptor desc; in Decode()
548 absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in Decode()
582 TensorLinearDescriptor desc; in Decode()
587 absl::make_unique<TensorLinearDescriptor>(std::move(desc))); in Decode()
664 dynamic_cast<const TensorLinearDescriptor*>(value.second.get()); in Encode()
717 dynamic_cast<const TensorLinearDescriptor*>(value.second.get()); in Encode()
|
| D | cl_arguments.cc | 186 const auto* linear_desc = dynamic_cast<const TensorLinearDescriptor*>(desc); in CreateCLObject()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | linear_storage.cc | 61 dynamic_cast<const TensorLinearDescriptor*>(obj_ptr); in GetGPUResources()
79 const TensorLinearDescriptor& desc, id<MTLDevice> device) { in CreateFromTensorLinearDescriptor()
|
| D | linear_storage.h | 52 const TensorLinearDescriptor& desc, id<MTLDevice> device);
|
| D | metal_arguments.cc | 143 const auto* linear_desc = dynamic_cast<const TensorLinearDescriptor*>(desc); in CreateMetalObject()
|