Searched refs:AddHalf (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | quantize_and_dequantize.cc | 47 op.args_.AddHalf("min", half(adjusted_attr.min)); in CreateQuantizeAndDequantize()
48 op.args_.AddHalf("max", half(adjusted_attr.max)); in CreateQuantizeAndDequantize()
49 op.args_.AddHalf("scale", half(adjusted_attr.scale)); in CreateQuantizeAndDequantize()
|
| D | relu.cc | 34 op.args_.AddHalf("alpha", half(attr.alpha)); in CreateReLU()
43 op.args_.AddHalf("clip", half(attr.clip)); in CreateReLU()
|
| D | prelu.cc | 74 result.args_.AddHalf("clip", half(attr.clip)); in CreatePReLU()
|
| D | elementwise.cc | 192 op.args_.AddHalf("scalar", half(scalar_parameter)); in CreateElementwiseOneRuntimeOneScalar()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | arguments.cc | 72 void Arguments::AddHalf(const std::string& name, half value) { in AddHalf() function in tflite::gpu::Arguments
131 AddHalf(RenameArg(object_names, postfix, v.first), v.second.value); in Merge()
|
| D | arguments.h | 60 void AddHalf(const std::string& name, half value = half(0.0f));
|