| /external/ComputeLibrary/arm_compute/graph/ |
| D | TensorDescriptor.h | 38 struct TensorDescriptor final : public misc::ICloneable<TensorDescriptor>
41 TensorDescriptor() = default;
50 TensorDescriptor(TensorShape tensor_shape,
64 TensorDescriptor &set_shape(TensorShape &tensor_shape) in set_shape()
75 TensorDescriptor &set_data_type(DataType tensor_data_type) in set_data_type()
86 TensorDescriptor &set_layout(DataLayout data_layout) in set_layout()
97 TensorDescriptor &set_quantization_info(QuantizationInfo tensor_quant_info) in set_quantization_info()
104 std::unique_ptr<TensorDescriptor> clone() const override in clone()
106 return std::make_unique<TensorDescriptor>(*this); in clone()
|
| D | Tensor.h | 49 Tensor(TensorID id, TensorDescriptor desc);
59 TensorDescriptor &desc();
64 const TensorDescriptor &desc() const;
115 TensorDescriptor _desc; /**< Tensor metadata */
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor.h | 44 Tensor(cl_mem memory, bool memory_owner, const TensorDescriptor& descriptor);
46 const TensorDescriptor& descriptor);
68 TensorDescriptor GetDescriptor() const override { return descriptor_; } in GetDescriptor()
83 absl::Status CreateFromDescriptor(const TensorDescriptor& desc,
85 absl::Status UploadDescriptorData(const TensorDescriptor& desc,
87 absl::Status ToDescriptor(TensorDescriptor* desc,
93 const TensorDescriptor& descriptor, int width_pixel_alignment,
105 TensorDescriptor descriptor_;
113 const TensorDescriptor& descriptor,
117 const TensorDescriptor& descriptor, Tensor* result);
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | tensor_desc.h | 44 class TensorDescriptor : public GPUObjectDescriptor {
46 TensorDescriptor() = default;
47 TensorDescriptor(DataType data_type, TensorStorageType storage_type, in TensorDescriptor() function
51 TensorDescriptor(const TensorDescriptor&) = default;
52 TensorDescriptor& operator=(const TensorDescriptor&) = default;
53 TensorDescriptor(TensorDescriptor&& desc);
54 TensorDescriptor& operator=(TensorDescriptor&& desc);
56 void CopyWithoutData(TensorDescriptor* desc) const;
58 bool operator==(const TensorDescriptor& d) const {
63 bool operator!=(const TensorDescriptor& d) const { return !(*this == d); }
[all …]
|
| D | testing_util.cc | 28 const std::vector<TensorDescriptor*>& src_cpu, in ExecuteGPUOperation()
29 const std::vector<TensorDescriptor*>& dst_cpu, in ExecuteGPUOperation()
57 std::vector<TensorDescriptor> src_cpu_descs(src_cpu.size()); in ExecuteGPUOperation()
58 std::vector<TensorDescriptor*> src_cpu_desc_ptrs(src_cpu.size()); in ExecuteGPUOperation()
64 std::vector<TensorDescriptor> dst_cpu_descs(dst_cpu.size()); in ExecuteGPUOperation()
65 std::vector<TensorDescriptor*> dst_cpu_desc_ptrs(dst_cpu.size()); in ExecuteGPUOperation()
87 std::vector<TensorDescriptor> src_cpu_descs(src_cpu.size()); in ExecuteGPUOperation()
88 std::vector<TensorDescriptor*> src_cpu_desc_ptrs(src_cpu.size()); in ExecuteGPUOperation()
94 std::vector<TensorDescriptor> dst_cpu_descs(dst_cpu.size()); in ExecuteGPUOperation()
95 std::vector<TensorDescriptor*> dst_cpu_desc_ptrs(dst_cpu.size()); in ExecuteGPUOperation()
[all …]
|
| D | tensor_desc.cc | 137 TensorDescriptor::TensorDescriptor(TensorDescriptor&& desc) in TensorDescriptor() function in tflite::gpu::TensorDescriptor
148 TensorDescriptor& TensorDescriptor::operator=(TensorDescriptor&& desc) { in operator =()
164 void TensorDescriptor::CopyWithoutData(TensorDescriptor* desc) const { in CopyWithoutData()
175 std::vector<uint64_t> TensorDescriptor::GetStorageDims() const { in GetStorageDims()
230 int3 TensorDescriptor::GetFullTensorRegion() const { in GetFullTensorRegion()
270 uint64_t TensorDescriptor::GetMemorySizeInBytes() const { in GetMemorySizeInBytes()
280 int TensorDescriptor::GetElementSize() const { in GetElementSize()
288 GPUResources TensorDescriptor::GetGPUResources(const GpuInfo& gpu_info) const { in GetGPUResources()
358 void TensorDescriptor::GetGpuResources( in GetGpuResources()
384 absl::Status TensorDescriptor::PerformConstExpr(const GpuInfo& gpu_info, in PerformConstExpr()
[all …]
|
| D | serialization_base.h | 33 flatbuffers::Offset<data::TensorDescriptor> Encode(
34 const TensorDescriptor& desc, flatbuffers::FlatBufferBuilder* builder);
35 void Decode(const data::TensorDescriptor* fb_desc, TensorDescriptor* desc);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor.h | 41 const TensorDescriptor& descriptor);
63 TensorDescriptor GetDescriptor() const override { return descriptor_; } in GetDescriptor()
72 absl::Status CreateFromDescriptor(const TensorDescriptor& desc,
74 absl::Status UploadDescriptorData(const TensorDescriptor& desc,
76 absl::Status ToDescriptor(TensorDescriptor* desc, id<MTLDevice> device) const;
83 id<MTLBuffer> buffer, const TensorDescriptor& descriptor,
87 id<MTLBuffer> buffer, const TensorDescriptor& descriptor,
100 TensorDescriptor descriptor_;
108 const TensorDescriptor& descriptor,
112 const TensorDescriptor& descriptor,
[all …]
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | FullyConnectedLayer.cpp | 55 TensorDescriptor FullyConnectedLayerNode::compute_weights_descriptor(const TensorDescriptor &input_… in compute_weights_descriptor()
72 TensorDescriptor weights_descriptor = input_descriptor; in compute_weights_descriptor()
90 TensorDescriptor FullyConnectedLayerNode::compute_output_descriptor(const TensorDescriptor &input_d… in compute_output_descriptor()
102 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
131 TensorDescriptor FullyConnectedLayerNode::configure_output(size_t idx) const in configure_output()
|
| D | StackLayerNode.cpp | 50 TensorDescriptor StackLayerNode::compute_output_descriptor(const std::vector<TensorDescriptor> &inp… in compute_output_descriptor()
55 TensorDescriptor output_descriptor = input_descriptors[0]; in compute_output_descriptor()
77 TensorDescriptor StackLayerNode::configure_output(size_t idx) const in configure_output()
88 TensorDescriptor output_info = {}; in configure_output()
92 std::vector<TensorDescriptor> inputs_descriptors; in configure_output()
|
| D | DepthToSpaceLayerNode.cpp | 47 TensorDescriptor DepthToSpaceLayerNode::compute_output_descriptor(const TensorDescriptor &input_des… in compute_output_descriptor()
55 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
73 TensorDescriptor DepthToSpaceLayerNode::configure_output(size_t idx) const in configure_output()
81 TensorDescriptor output_desc = src->desc(); in configure_output()
|
| D | DeconvolutionLayerNode.cpp | 47 TensorDescriptor DeconvolutionLayerNode::compute_output_descriptor(const TensorDescriptor &input_de… in compute_output_descriptor()
48 … const TensorDescriptor &weights_descriptor, in compute_output_descriptor()
62 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
82 TensorDescriptor DeconvolutionLayerNode::configure_output(size_t idx) const in configure_output()
90 …TensorDescriptor output_info = compute_output_descriptor(src->desc(), weights->desc(), descriptor.… in configure_output()
|
| D | ConcatenateLayerNode.cpp | 64 TensorDescriptor ConcatenateLayerNode::compute_output_descriptor(const std::vector<TensorDescriptor… in compute_output_descriptor()
69 TensorDescriptor output_descriptor = input_descriptors[0]; in compute_output_descriptor()
98 TensorDescriptor ConcatenateLayerNode::configure_output(size_t idx) const in configure_output()
109 TensorDescriptor output_info = {}; in configure_output()
113 std::vector<TensorDescriptor> inputs_descriptors; in configure_output()
|
| D | FusedConvolutionBatchNormalizationWithPostOpsNode.cpp | 80 TensorDescriptor FusedConvolutionBatchNormalizationWithPostOpsNode::compute_output_descriptor(const… in compute_output_descriptor()
81 … const TensorDescriptor &weights_descriptor, in compute_output_descriptor()
95 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
115 TensorDescriptor FusedConvolutionBatchNormalizationWithPostOpsNode::configure_output(size_t idx) co… in configure_output()
123 TensorDescriptor output_info = compute_output_descriptor(src->desc(), weights->desc(), _info); in configure_output()
|
| D | FusedDepthwiseConvolutionBatchNormalizationNode.cpp | 81 TensorDescriptor FusedDepthwiseConvolutionBatchNormalizationNode::compute_output_descriptor(const T… in compute_output_descriptor()
82 … const TensorDescriptor &weights_descriptor, in compute_output_descriptor()
97 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
117 TensorDescriptor FusedDepthwiseConvolutionBatchNormalizationNode::configure_output(size_t idx) const in configure_output()
125 …TensorDescriptor output_info = compute_output_descriptor(src->desc(), weights->desc(), _info, _dep… in configure_output()
|
| D | ConvolutionLayerNode.cpp | 91 TensorDescriptor ConvolutionLayerNode::compute_output_descriptor(const TensorDescriptor &input_desc… in compute_output_descriptor()
92 … const TensorDescriptor &weights_descriptor, in compute_output_descriptor()
106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
126 TensorDescriptor ConvolutionLayerNode::configure_output(size_t idx) const in configure_output()
134 TensorDescriptor output_info = compute_output_descriptor(src->desc(), weights->desc(), _info); in configure_output()
|
| D | FusedConvolutionBatchNormalizationNode.cpp | 91 TensorDescriptor FusedConvolutionBatchNormalizationNode::compute_output_descriptor(const TensorDesc… in compute_output_descriptor()
92 … const TensorDescriptor &weights_descriptor, in compute_output_descriptor()
106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
126 TensorDescriptor FusedConvolutionBatchNormalizationNode::configure_output(size_t idx) const in configure_output()
134 TensorDescriptor output_info = compute_output_descriptor(src->desc(), weights->desc(), _info); in configure_output()
|
| D | DepthwiseConvolutionLayerNode.cpp | 78 TensorDescriptor DepthwiseConvolutionLayerNode::compute_output_descriptor(const TensorDescriptor &i… in compute_output_descriptor()
79 … const TensorDescriptor &weights_descriptor, in compute_output_descriptor()
95 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
115 TensorDescriptor DepthwiseConvolutionLayerNode::configure_output(size_t idx) const in configure_output()
123 …TensorDescriptor output_info = compute_output_descriptor(src->desc(), weights->desc(), _info, _dep… in configure_output()
|
| D | FusedConvolutionWithPostOpNode.cpp | 91 TensorDescriptor FusedConvolutionWithPostOpNode::compute_output_descriptor(const TensorDescriptor &… in compute_output_descriptor()
92 … const TensorDescriptor &weights_descriptor, in compute_output_descriptor()
106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
126 TensorDescriptor FusedConvolutionWithPostOpNode::configure_output(size_t idx) const in configure_output()
134 TensorDescriptor output_info = compute_output_descriptor(src->desc(), weights->desc(), _info); in configure_output()
|
| /external/ComputeLibrary/examples/ |
| D | graph_edsr.h | 51 TensorDescriptor in setup()
63 TensorDescriptor in setup()
74 TensorDescriptor in setup()
86 TensorDescriptor in setup()
97 TensorDescriptor in setup()
109 TensorDescriptor in setup()
120 TensorDescriptor in setup()
131 TensorDescriptor in setup()
143 TensorDescriptor in setup()
154 TensorDescriptor in setup()
[all …]
|
| /external/ComputeLibrary/arm_compute/graph/nodes/ |
| D | FullyConnectedLayerNode.h | 74 static TensorDescriptor compute_weights_descriptor(const TensorDescriptor &input_descriptor,
88 static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor,
100 TensorDescriptor configure_output(size_t idx) const override;
|
| D | DeconvolutionLayerNode.h | 56 static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor,
57 const TensorDescriptor &weights_descriptor,
63 TensorDescriptor configure_output(size_t idx) const override;
|
| D | FusedDepthwiseConvolutionBatchNormalizationNode.h | 93 static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor,
94 const TensorDescriptor &weights_descriptor,
119 TensorDescriptor configure_output(size_t idx) const override;
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | subgraph.h | 43 std::vector<TensorDescriptor> new_tensors;
45 int AddTensor(const TensorDescriptor& desc);
46 int AddTensor(const BHWC& shape, const TensorDescriptor& desc);
47 int AddTensor(const OHWI& shape, const TensorDescriptor& desc);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | prelu.cc | 30 TensorDescriptor tensor_desc) { in CreatePReLU()
36 TensorDescriptor alpha_tensor_desc = CreateConstantLinearTensorDescriptor( in CreatePReLU()
38 op_desc.args.AddObject("alpha", std::make_unique<TensorDescriptor>( in CreatePReLU()
48 TensorDescriptor const_tensor_desc = tensor_desc; in CreatePReLU()
50 op_desc.args.AddObject("alpha", std::make_unique<TensorDescriptor>( in CreatePReLU()
|