Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.h47 std::vector<ValueId> inputs;
48 std::vector<ValueId> outputs;
85 absl::Status SetInputTensor(ValueId id, const TensorFloat32& tensor,
90 Tensor* GetTensor(ValueId id);
92 absl::Status GetOutputTensor(ValueId id, CLCommandQueue* queue,
95 const std::vector<ValueId>& GetInputIds() const { return input_ids_; } in GetInputIds()
96 const std::vector<ValueId>& GetOutputIds() const { return output_ids_; } in GetOutputIds()
131 void GetUsages(const std::function<bool(ValueId)>& functor,
132 std::map<ValueId, int2>* usages);
134 TensorMemoryType GetTensorMemoryType(ValueId id);
[all …]
Dinference_context.cc53 bool IsReady(const absl::flat_hash_set<ValueId>& ready_tensors, in IsReady()
55 for (const ValueId in_id : node.inputs) { in IsReady()
63 std::vector<std::pair<ValueId, TensorDescriptor>> GetCLNodeTensors( in GetCLNodeTensors()
65 std::vector<std::pair<ValueId, TensorDescriptor>> result; in GetCLNodeTensors()
87 void AddUsage(ValueId id, int task_index, in AddUsage()
88 std::map<ValueId, int2>* usage_records) { in AddUsage() argument
266 ValueId max_id = 0; in ReserveGraphTensors()
294 std::map<ValueId, TensorDescriptor> tensor_descriptors; in ConvertOperations()
301 std::map<ValueId, int> in ConvertOperations()
368 absl::flat_hash_map<int, ValueId> mapping_to_global_ids; in ConvertOperations()
[all …]
Dserialization.cc901 const TensorDescriptor& desc, const ValueId& id, in Encode()
911 ValueId* id) { in Decode()
1037 std::vector<std::pair<ValueId, TensorDescriptor>> tensors; in Decode()
Dapi.cc782 std::vector<TensorTieDef> LinkTensors(const std::vector<ValueId>& ids, in LinkTensors()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.h43 std::vector<ValueId> inputs;
44 std::vector<ValueId> outputs;
83 const std::map<ValueId, id<MTLBuffer>>& preallocated);
133 const std::set<ValueId>& preallocated_ids);
139 const std::set<ValueId>& preallocated_ids);
145 MetalSpatialTensor* GetTensor(ValueId tensor_id);
146 void GetUsages(const std::function<bool(ValueId)>& functor,
147 std::map<ValueId, int2>* usages);
148 TensorMemoryType GetTensorMemoryType(ValueId id);
163 ValueId Add(const DummyTensor& dummy) { in Add()
[all …]
Dinference_context.cc53 bool HasIntersection(const std::vector<ValueId>& vec_ids, in HasIntersection()
54 const std::set<ValueId>& ids) { in HasIntersection()
55 for (ValueId id : vec_ids) { in HasIntersection()
63 bool IsReady(const std::set<ValueId>& ready_tensors, const MetalNode& node) { in IsReady()
64 for (const ValueId in_id : node.inputs) { in IsReady()
72 void AddUsage(ValueId id, int task_index, in AddUsage()
73 std::map<ValueId, int2>* usage_records) { in AddUsage() argument
137 std::set<ValueId> preallocated_ids; in InitFromGraph()
166 const GraphFloat32& graph, const std::set<ValueId>& preallocated_ids) { in ReserveGraphTensors()
167 ValueId max_id = 0; in ReserveGraphTensors()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel.h39 using ValueId = uint32_t; variable
52 const ValueId id;
102 bool IsGraphInput(ValueId id) const;
104 bool IsGraphOutput(ValueId id) const;
107 Node* FindProducer(ValueId id) const;
111 std::vector<Node*> FindConsumers(ValueId id) const;
117 Value* GetValue(ValueId id) const;
138 absl::Status SetProducer(NodeId producer, ValueId value);
142 absl::Status RemoveProducer(ValueId value);
146 absl::Status AddConsumer(NodeId consumer, ValueId value);
[all …]
Dmodel.cc72 bool GraphFloat32::IsGraphInput(ValueId id) const { in IsGraphInput()
79 bool GraphFloat32::IsGraphOutput(ValueId id) const { in IsGraphOutput()
86 Node* GraphFloat32::FindProducer(ValueId id) const { in FindProducer()
93 std::vector<Node*> GraphFloat32::FindConsumers(ValueId id) const { in FindConsumers()
107 Value* GraphFloat32::GetValue(ValueId id) const { in GetValue()
149 absl::make_unique<Value>(Value{static_cast<ValueId>(values_.size()), {}}); in NewValue()
155 absl::Status GraphFloat32::SetProducer(NodeId producer, ValueId value) { in SetProducer()
183 absl::Status GraphFloat32::RemoveProducer(ValueId value) { in RemoveProducer()
195 absl::Status GraphFloat32::AddConsumer(NodeId consumer, ValueId value) { in AddConsumer()
220 absl::Status GraphFloat32::ReplaceInput(NodeId node, ValueId old_value, in ReplaceInput()
[all …]
Dlstm_parser.h30 absl::flat_hash_map<int, ValueId>* new_variable_input_values);
Doperation_parser.h47 virtual absl::flat_hash_map<int, ValueId>
Dmodel_builder.cc942 absl::flat_hash_map<int, ValueId> GetNewValueIdsForVariableInputNodes() in GetNewValueIdsForVariableInputNodes()
1036 absl::flat_hash_map<int, ValueId> new_variable_input_value_map_;
2560 const absl::flat_hash_map<int, ValueId>& new_variable_tensor_values) { in CopyVariableTensorOutputs()
2561 absl::flat_hash_map<int, ValueId> new_variable_tensor_values_copy( in CopyVariableTensorOutputs()
2625 std::vector<ValueId> variable_inputs_to_value_id; in BuildModel()
2647 absl::flat_hash_map<int, ValueId> new_value_for_variable_input_tensors = in BuildModel()
Dlstm_parser.cc446 absl::flat_hash_map<int, ValueId>* new_variable_input_values) { in ParseLSTMAttributes()
/external/deqp/external/vulkancts/modules/vulkan/spirv_assembly/
DvktSpvAsmFloatControlsTests.cpp116 enum ValueId enum
304 ValueId opVarResult;
305 ValueId opDenormResult;
306 ValueId opInfResult;
307 ValueId opNanResult;
312 ValueId result;
368 virtual BufferSp constructInputBuffer (const ValueId* twoArguments) const = 0;
369 virtual BufferSp constructOutputBuffer (ValueId result) const = 0;
370 …virtual void fillInputData (const ValueId* twoArguments, vector<deUint8>& bufferData, deUint32&…
389 BufferSp constructInputBuffer (const ValueId* twoArguments) const override;
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/
Dfuse_auto_input.cc60 std::vector<std::pair<ValueId, int>> input_values; in ApplyToNode()
64 const ValueId input_id = input_value->id; in ApplyToNode()
105 absl::flat_hash_set<ValueId> all_inputs; in ApplyToNode()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dgl_delegate.cc288 for (ValueId id : inputs_) { in Invoke()
312 for (ValueId id : outputs_) { in Invoke()
342 absl::Status FindObject(ValueId id, ValueRef* ref) const { in FindObject()
363 std::vector<ValueId> inputs_;
364 std::vector<ValueId> outputs_;
Dmetal_delegate.mm361 std::vector<::tflite::gpu::ValueId> input_ids;
363 std::map<::tflite::gpu::ValueId, BHWC> input_dimensions;
365 for (const ValueId input : inputs_) {
402 std::vector<::tflite::gpu::ValueId> output_ids;
405 for (const ValueId output : outputs_) {
591 std::vector<ValueRef> tensors_; // indexed by ValueId
592 std::vector<ValueId> inputs_;
593 std::vector<ValueId> outputs_;
603 std::map<::tflite::gpu::ValueId, id<MTLBuffer>> input_output_buffers_;
604 std::map<::tflite::gpu::ValueId, id<MTLBuffer>> bphwc4_buffers_;
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dspecial_selector.cc34 const std::map<ValueId, TensorDescriptor>& tensor_descriptors, in TryDepthwiseConvPlus1x1Conv() argument
98 const std::map<ValueId, TensorDescriptor>& tensor_descriptors, in TryFCFCAdd() argument
188 const std::map<ValueId, TensorDescriptor>& tensor_descriptors, in GPUSubgraphFromGraph() argument
Dspecial_selector.h36 const std::map<ValueId, TensorDescriptor>& tensor_descriptors,
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dapi.cc49 using ObjectsSizes = absl::flat_hash_map<ValueId, size_t>;
108 const ValueId id = s.first; in Execute()
132 const ValueId id = s.first; in Execute()
371 absl::flat_hash_map<ValueId, size_t> object_sizes_;
Dcompiler.cc163 absl::flat_hash_map<ValueId, Object> objects; in Compile()
/external/llvm/lib/Bitcode/Reader/
DBitcodeReader.cpp546 getGUIDFromValueId(unsigned ValueId);
5855 ModuleSummaryIndexBitcodeReader::getGUIDFromValueId(unsigned ValueId) { in getGUIDFromValueId() argument
5856 auto VGI = ValueIdToCallGraphGUIDMap.find(ValueId); in getGUIDFromValueId()
5968 unsigned ValueId = 0; in parseModule() local
6077 ValueIdToLinkageMap[ValueId++] = Linkage; in parseModule()
6088 ValueIdToLinkageMap[ValueId++] = Linkage; in parseModule()
6098 ValueIdToLinkageMap[ValueId++] = Linkage; in parseModule()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Bitcode/Reader/
DBitcodeReader.cpp836 getValueInfoFromValueId(unsigned ValueId);
5379 ModuleSummaryIndexBitcodeReader::getValueInfoFromValueId(unsigned ValueId) { in getValueInfoFromValueId() argument
5380 auto VGI = ValueIdToValueInfoMap[ValueId]; in getValueInfoFromValueId()
5510 unsigned ValueId = 0; in parseModule() local
5629 ValueIdToLinkageMap[ValueId++] = Linkage; in parseModule()
5633 setValueGUID(ValueId++, Name, Linkage, SourceFileName); in parseModule()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Bitcode/Writer/
DBitcodeWriter.cpp4000 auto ValueId = getValueId(I.first); in writeCombinedGlobalValueSummary() local
4001 assert(ValueId); in writeCombinedGlobalValueSummary()
4002 SummaryToValueIdMap[S] = *ValueId; in writeCombinedGlobalValueSummary()
4018 NameVals.push_back(*ValueId); in writeCombinedGlobalValueSummary()
4041 NameVals.push_back(*ValueId); in writeCombinedGlobalValueSummary()
/external/llvm/lib/Bitcode/Writer/
DBitcodeWriter.cpp3437 unsigned ValueId = getValueId(I.first); in writeCombinedGlobalValueSummary() local
3438 SummaryToValueIdMap[S] = ValueId; in writeCombinedGlobalValueSummary()
3448 NameVals.push_back(ValueId); in writeCombinedGlobalValueSummary()
3464 NameVals.push_back(ValueId); in writeCombinedGlobalValueSummary()
/external/llvm-project/llvm/lib/Bitcode/Reader/
DBitcodeReader.cpp840 getValueInfoFromValueId(unsigned ValueId);
5583 ModuleSummaryIndexBitcodeReader::getValueInfoFromValueId(unsigned ValueId) { in getValueInfoFromValueId() argument
5584 auto VGI = ValueIdToValueInfoMap[ValueId]; in getValueInfoFromValueId()
5714 unsigned ValueId = 0; in parseModule() local
5833 ValueIdToLinkageMap[ValueId++] = Linkage; in parseModule()
5837 setValueGUID(ValueId++, Name, Linkage, SourceFileName); in parseModule()

12