MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetArgsDeclaration

Method GetArgsDeclaration

tensorflow/lite/delegates/gpu/cl/kernels/add.cc:158–170  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

156}
157
158std::string Add::GetArgsDeclaration() const {
159 std::string args;
160 for (int i = 1; i < src_depthes_.size(); ++i) {
161 const std::string tensor_name =
162 absl::StrCat("src_data_", link_index_, "_", i);
163 TensorCodeGenerator src_tensor(tensor_name, "", definition_.src_tensors[i]);
164 absl::StrAppend(&args, ",\n", src_tensor.GetDeclaration(AccessType::READ));
165 }
166 if (HasTexture2DStorageType(definition_)) {
167 absl::StrAppend(&args, ",\n float ", inv_divisor_name_);
168 }
169 return args;
170}
171
172Status Add::BindArguments(CLKernel* kernel) {
173 for (int i = 1; i < src_depthes_.size(); ++i) {

Callers 2

GetElementWiseCodeFunction · 0.45
GetArgsDeclarationFunction · 0.45

Calls 5

HasTexture2DStorageTypeFunction · 0.85
StrCatFunction · 0.50
StrAppendFunction · 0.50
sizeMethod · 0.45
GetDeclarationMethod · 0.45

Tested by

no test coverage detected