Home
last modified time | relevance | path

Searched refs:hfvalue (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_arguments.cc405 for (const auto& hfvalue : half_values_) { in CopyScalarValues() local
406 args->half_values_[hfvalue.first].value = hfvalue.second.value; in CopyScalarValues()
429 for (const auto& hfvalue : args.half_values_) { in CopyArguments() local
430 auto& new_val = half_values_[hfvalue.first]; in CopyArguments()
431 new_val.value = hfvalue.second.value; in CopyArguments()
432 new_val.active = hfvalue.second.active; in CopyArguments()
433 if (hfvalue.second.active) { in CopyArguments()
471 for (const auto& hfvalue : half_values_) { in RenameArgumentsInCode() local
472 if (hfvalue.second.active) { in RenameArgumentsInCode()
473 std::string index = std::to_string(hfvalue.second.offset / 4); in RenameArgumentsInCode()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_arguments.cc247 for (const auto& hfvalue : args->half_values_) { in ScalarArgumentsToStructWithScalarFields() local
248 auto& new_val = float_values_[hfvalue.first]; in ScalarArgumentsToStructWithScalarFields()
249 new_val.value = hfvalue.second.value; in ScalarArgumentsToStructWithScalarFields()
250 new_val.active = hfvalue.second.active; in ScalarArgumentsToStructWithScalarFields()
251 if (hfvalue.second.active) { in ScalarArgumentsToStructWithScalarFields()
254 struct_desc += " float " + hfvalue.first + ";\n"; in ScalarArgumentsToStructWithScalarFields()
255 ReplaceAllWords(kArgsPrefix + hfvalue.first, in ScalarArgumentsToStructWithScalarFields()
256 "static_cast<half>(U." + hfvalue.first + ")", code); in ScalarArgumentsToStructWithScalarFields()
317 for (const auto& hfvalue : args->half_values_) { in ScalarArgumentsToStructWithVec4Fields() local
318 auto& new_val = float_values_[hfvalue.first]; in ScalarArgumentsToStructWithVec4Fields()
[all …]