Home
last modified time | relevance | path

Searched refs:ValueId (Results 1 – 25 of 63) sorted by relevance

123

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.h44 std::vector<ValueId> inputs;
45 std::vector<ValueId> outputs;
82 Status SetInputTensor(ValueId id, const TensorFloat32& tensor,
87 Tensor* GetTensor(ValueId id);
89 Status GetOutputTensor(ValueId id, CLCommandQueue* queue,
110 std::map<ValueId, int2>* usages);
146 ValueId Add(const DummyTensor& dummy) { in Add()
150 void Add(ValueId id, const DummyTensor& dummy) { in Add()
153 void SetNext(ValueId id) { next_ = id; } in SetNext()
154 DummyTensor Get(ValueId id) { return reservations_[id]; } in Get()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompiled_model.cc49 std::vector<ValueId> input_buffer_ids;
51 std::vector<ValueId> unused_input_buffer_ids;
53 std::vector<ValueId> extra_output_buffer_ids;
55 std::vector<ValueId> missing_output_buffer_ids;
60 bool Contains(const std::vector<ValueId>& container, ValueId value) { in Contains()
66 bool Contains(const std::vector<T>& container, ValueId value) { in Contains()
77 bool Contains(const std::vector<ValueId>& wide, in Contains()
78 const std::vector<ValueId>& narrow) { in Contains()
82 std::set<ValueId> wide_sorted; in Contains()
95 const std::vector<ValueId>& wide, in Contains()
[all …]
Dinference_context_test.mm37 using ::tflite::gpu::ValueId;
40 static std::vector<ComputeTaskDescriptorPtr> MulLinkable(int id, ValueId input_id,
41 ValueId output_id) {
54 static std::vector<ComputeTaskDescriptorPtr> Add(int id, ValueId input_id, ValueId output_id) {
80 [input_id, output_id](const std::map<ValueId, BHWC>& buffers) {
86 [output_id](const std::map<ValueId, BHWC>& buffers) {
96 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) {
109 int id, ValueId input_id, ValueId output_id, const std::vector<float>& channel_multipliers) {
121 [channel_multipliers](const std::map<ValueId, BHWC>& buffers) {
130 int id, ValueId input_id, ValueId output_id, const std::vector<float>& channel_multipliers) {
[all …]
Dapi.cc55 const GraphFloat32& graph, int id, ValueId input_id, ValueId output_id, in SelectConvolution()
87 int id, ValueId input_id, ValueId output_id, in SelectDepthWiseConv()
100 const GraphFloat32& graph, int id, ValueId input_id, ValueId output_id, in SelectPReLU()
119 const GraphFloat32& graph, int id, ValueId input_id, ValueId output_id, in SelectReshape()
130 int id, ValueId input_id, in SelectSoftmax()
131 ValueId output_id) { in SelectSoftmax()
141 const std::vector<ValueId>& inputs, in RegisterPrimaryOps()
142 const std::vector<ValueId>& outputs, in RegisterPrimaryOps()
292 std::vector<ValueId> inputs; in Compile()
294 inputs.push_back(static_cast<ValueId>(input->id)); in Compile()
[all …]
Dcompute_task_descriptor.h36 std::function<BHWC(const std::map<ValueId, BHWC>& buffers)>;
38 std::function<std::vector<uint8_t>(const std::map<ValueId, BHWC>& buffers)>;
40 const std::map<ValueId, BHWC>& buffers)>;
49 ValueId id;
56 ValueId id;
67 std::vector<ValueId> alias;
Dcompute_task.h41 dimensions:(std::map<::tflite::gpu::ValueId, ::tflite::gpu::BHWC>*)dimensions;
53 - (::tflite::gpu::Status)assignBuffers:(std::map<::tflite::gpu::ValueId, id<MTLBuffer>>*)buffers
54 outputIds:(const std::vector<::tflite::gpu::ValueId>&)outputIds
56 (const std::map<::tflite::gpu::ValueId, size_t>&)usageRecordIds
62 (const std::map<::tflite::gpu::ValueId, id<MTLBuffer>>&)inputOutputBuffers;
Dcompiled_model_test.mm30 using ::tflite::gpu::ValueId;
33 static std::vector<ComputeTaskDescriptorPtr> MulLinkable(int id, ValueId input_id,
34 ValueId output_id) {
47 static std::vector<ComputeTaskDescriptorPtr> Add(int id, ValueId input_id, ValueId output_id) {
73 [input_id, output_id](const std::map<ValueId, BHWC>& buffers) {
79 [output_id](const std::map<ValueId, BHWC>& buffers) {
89 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) {
101 static std::vector<ComputeTaskDescriptorPtr> Add2(int id, ValueId input_id1, ValueId input_id2,
102 ValueId output_id) {
129 [input_id1](const std::map<ValueId, BHWC>& buffers) {
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dconv.h31 int id, ValueId input_id, ValueId output_id,
45 int id, ValueId input_id, ValueId output_id,
57 int id, ValueId input_id, ValueId output_id,
71 int id, ValueId input_id, ValueId output_id,
79 int id, ValueId input_id, ValueId output_id,
Dsoftmax.cc99 std::vector<ComputeTaskDescriptorPtr> Softmax(int id, ValueId input_id, in Softmax()
100 ValueId output_id, in Softmax()
148 [input_id](const std::map<ValueId, BHWC>& buffers) { in Softmax()
154 [output_id](const std::map<ValueId, BHWC>& buffers) { in Softmax()
161 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in Softmax() argument
172 std::vector<ComputeTaskDescriptorPtr> Softmax1x1(int id, ValueId input_id, in Softmax1x1()
173 ValueId output_id, in Softmax1x1()
185 [input_id](const std::map<ValueId, BHWC>& buffers) { in Softmax1x1()
191 [channels_count](const std::map<ValueId, BHWC>& buffers) { in Softmax1x1()
209 desc->resize_function = [](const std::map<ValueId, BHWC>& buffers) { in Softmax1x1() argument
Ddepthwise_conv.h31 int id, ValueId input_id, ValueId output_id,
42 int id, ValueId input_id, ValueId output_id,
57 int id, ValueId input_id, ValueId output_id,
Dmul.cc82 std::vector<ComputeTaskDescriptorPtr> ApplyMask(int id, ValueId input_id_0, in ApplyMask()
83 ValueId input_id_1, in ApplyMask()
84 ValueId output_id, in ApplyMask()
98 [input_id_0, input_id_1](const std::map<ValueId, BHWC>& buffers) { in ApplyMask()
105 output_id](const std::map<ValueId, BHWC>& buffers) { in ApplyMask()
119 input_id_1](const std::map<ValueId, BHWC>& buffers) { in ApplyMask() argument
131 std::vector<ComputeTaskDescriptorPtr> Multiply(int id, ValueId input_id, in Multiply()
132 ValueId output_id, in Multiply()
161 [multiplier_bits](const std::map<ValueId, BHWC>& buffers) { in Multiply()
Dconcat.cc113 int id, std::vector<ValueId> input_ids, ValueId output_id, in ConcatZ()
133 [input_ids, attr](const std::map<ValueId, BHWC>& buffers) { in ConcatZ()
145 [input_ids](const std::map<ValueId, BHWC>& buffers) { in ConcatZ()
157 desc->resize_function = [input_ids](const std::map<ValueId, BHWC>& buffers) { in ConcatZ() argument
170 int id, std::vector<ValueId> input_ids, ValueId output_id, in ConcatX()
226 [input_ids, attr](const std::map<ValueId, BHWC>& buffers) { in ConcatX()
238 [output_id](const std::map<ValueId, BHWC>& buffers) { in ConcatX()
247 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in ConcatX() argument
260 int id, std::vector<ValueId> input_ids, ValueId output_id, in ConcatY()
317 [input_ids, attr](const std::map<ValueId, BHWC>& buffers) { in ConcatY()
[all …]
Dmul.h29 std::vector<ComputeTaskDescriptorPtr> Multiply(int id, ValueId input_id,
30 ValueId output_id,
34 std::vector<ComputeTaskDescriptorPtr> ApplyMask(int id, ValueId input_id_0,
35 ValueId input_id_1,
36 ValueId output_id,
Dreshape.cc119 std::vector<ComputeTaskDescriptorPtr> Reshape(int id, ValueId input_id, in Reshape()
120 ValueId output_id, in Reshape()
133 [input_id, attr](const std::map<ValueId, BHWC>& buffers) { in Reshape()
141 [input_id, output_id](const std::map<ValueId, BHWC>& buffers) { in Reshape()
160 desc->resize_function = [attr](const std::map<ValueId, BHWC>& buffers) { in Reshape() argument
173 std::vector<ComputeTaskDescriptorPtr> Reshapex4(int id, ValueId input_id, in Reshapex4()
174 ValueId output_id, in Reshapex4()
187 [input_id, attr](const std::map<ValueId, BHWC>& buffers) { in Reshapex4()
195 [input_id, output_id](const std::map<ValueId, BHWC>& buffers) { in Reshapex4()
219 desc->resize_function = [attr](const std::map<ValueId, BHWC>& buffers) { in Reshapex4() argument
Dprelu.cc37 std::vector<ComputeTaskDescriptorPtr> PReLU(int id, ValueId input_id, in PReLU()
38 ValueId output_id, in PReLU()
71 [attr](const std::map<ValueId, BHWC>& buffers) { in PReLU()
81 std::vector<ComputeTaskDescriptorPtr> PReLUFull(int id, ValueId input_id, in PReLUFull()
82 ValueId output_id, in PReLUFull()
114 [attr](const std::map<ValueId, BHWC>& buffers) { in PReLUFull()
Ddepthwise_conv.cc468 int id, ValueId input_id, ValueId output_id, in DepthWiseConvolution()
563 [input_id, attr](const std::map<ValueId, BHWC>& buffers) { in DepthWiseConvolution()
581 [input_id, output_id, attr](const std::map<ValueId, BHWC>& buffers) { in DepthWiseConvolution()
610 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in DepthWiseConvolution() argument
623 int id, ValueId input_id, ValueId output_id, in DepthWiseConv3x3Stride1x1()
637 [input_id, attr](const std::map<ValueId, BHWC>& buffers) { in DepthWiseConv3x3Stride1x1()
652 [input_id, output_id, attr](const std::map<ValueId, BHWC>& buffers) { in DepthWiseConv3x3Stride1x1()
660 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in DepthWiseConv3x3Stride1x1() argument
687 int id, ValueId input_id, ValueId output_id, in DepthWiseConv3x3Stride2()
701 [input_id, attr](const std::map<ValueId, BHWC>& buffers) { in DepthWiseConv3x3Stride2()
[all …]
Delementwise.cc94 int id, std::vector<ValueId> input_ids, ValueId output_id, in ElementwiseWithTwoInputs()
109 [input_ids](const std::map<ValueId, BHWC>& buffers) { in ElementwiseWithTwoInputs()
115 [input_ids](const std::map<ValueId, BHWC>& buffers) { in ElementwiseWithTwoInputs()
122 desc->resize_function = [input_ids](const std::map<ValueId, BHWC>& buffers) { in ElementwiseWithTwoInputs() argument
135 int id, ValueId input_id, ValueId output_id, OperationType op_type) { in ElementwiseWithOneInput()
Dmax_unpooling.cc90 int id, ValueId input_id, ValueId input_indices_id, ValueId output_id, in MaxUnpooling()
104 params](const std::map<ValueId, BHWC>& buffers) { in MaxUnpooling()
112 params](const std::map<ValueId, BHWC>& buffers) { in MaxUnpooling()
130 params](const std::map<ValueId, BHWC>& buffers) { in MaxUnpooling() argument
Dsoftmax.h30 std::vector<ComputeTaskDescriptorPtr> Softmax(int id, ValueId input_id,
31 ValueId output_id,
36 std::vector<ComputeTaskDescriptorPtr> Softmax1x1(int id, ValueId input_id,
37 ValueId output_id,
Dreshape.h30 std::vector<ComputeTaskDescriptorPtr> Reshape(int id, ValueId input_id,
31 ValueId output_id,
36 std::vector<ComputeTaskDescriptorPtr> Reshapex4(int id, ValueId input_id,
37 ValueId output_id,
Delementwise.h29 int id, std::vector<ValueId> input_ids, ValueId output_id,
33 int id, ValueId input_id, ValueId output_id, OperationType op_type);
Dtest_util.mm69 std::vector<ValueId> input_ids;
74 std::vector<ValueId> output_ids;
94 std::map<ValueId, BHWC> input_dimensions;
95 std::map<ValueId, id<MTLBuffer>> input_buffers;
112 std::map<ValueId, BHWC> output_dimensions;
117 std::map<ValueId, id<MTLBuffer>> output_buffers;
120 const ValueId key = outputDimension.first;
127 std::map<ValueId, id<MTLBuffer>> inout_buffers(input_buffers.begin(), input_buffers.end());
170 const std::map<ValueId, TensorFloat32>& inputs,
171 std::map<ValueId, TensorFloat32>* outputs) {
[all …]
Dresize.cc90 std::vector<ComputeTaskDescriptorPtr> Resize(int id, ValueId input_id, in Resize()
91 ValueId output_id, in Resize()
114 [input_id, attr](const std::map<ValueId, BHWC>& buffers) { in Resize()
120 [input_id, output_id](const std::map<ValueId, BHWC>& buffers) { in Resize()
132 [input_id, output_id, attr](const std::map<ValueId, BHWC>& buffers) { in Resize()
145 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in Resize() argument
Dtranspose_conv.h31 int id, ValueId input_id, ValueId output_id,
36 int id, ValueId input_id, ValueId output_id,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel.h38 using ValueId = uint32_t; variable
47 const ValueId id;
96 virtual bool IsGraphInput(ValueId id) const = 0;
98 virtual bool IsGraphOutput(ValueId id) const = 0;
101 virtual Node* FindProducer(ValueId id) const = 0;
105 virtual std::vector<Node*> FindConsumers(ValueId id) const = 0;
111 virtual Value<TensorT>* GetValue(ValueId id) const = 0;
129 virtual Status SetProducer(NodeId producer, ValueId value) = 0;
133 virtual Status RemoveProducer(ValueId value) = 0;
137 virtual Status AddConsumer(NodeId consumer, ValueId value) = 0;
[all …]

123