Searched refs:ElementwiseDescriptor (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | elementwise.cc | 231 ElementwiseDescriptor CreateElementwiseOneRuntimeOneScalar( in CreateElementwiseOneRuntimeOneScalar()
234 ElementwiseDescriptor op_desc; in CreateElementwiseOneRuntimeOneScalar()
248 ElementwiseDescriptor CreateElementwiseTwoInput( in CreateElementwiseTwoInput()
255 ElementwiseDescriptor op_desc; in CreateElementwiseTwoInput()
274 ElementwiseDescriptor CreateElementwiseTwoInput( in CreateElementwiseTwoInput()
285 ElementwiseDescriptor op_desc; in CreateElementwiseTwoInput()
305 ElementwiseDescriptor CreateElementwiseDesc(const GpuInfo& gpu_info, in CreateElementwiseDesc()
326 return ElementwiseDescriptor(); in CreateElementwiseDesc()
335 ElementwiseDescriptor op_desc; in CreateElementwiseOneInput()
352 ElementwiseDescriptor op_desc; in CreateElementwiseTwoInput()
[all …]
|
| D | relu.cc | 26 ElementwiseDescriptor CreateReLU(const ReLUAttributes& attr, in CreateReLU()
28 ElementwiseDescriptor result; in CreateReLU()
56 ElementwiseDescriptor op_desc = CreateReLU(attr, definition.precision); in CreateReLU()
|
| D | prelu.cc | 29 ElementwiseDescriptor CreatePReLU(const PReLUAttributes& attr, in CreatePReLU()
31 ElementwiseDescriptor op_desc; in CreatePReLU()
74 ElementwiseDescriptor op_desc = CreatePReLU(attr, definition.src_tensors[0]); in CreatePReLU()
|
| D | relu.h | 27 ElementwiseDescriptor CreateReLU(const ReLUAttributes& attr,
|
| D | prelu.h | 30 ElementwiseDescriptor CreatePReLU(const PReLUAttributes& attr,
|
| D | cast.cc | 30 ElementwiseDescriptor op_desc; in CreateCast()
|
| D | quantize_and_dequantize.cc | 41 ElementwiseDescriptor op_desc; in CreateQuantizeAndDequantize()
|
| D | add.cc | 73 ElementwiseDescriptor op_desc; in CreateAdd()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | gpu_operation.h | 80 struct ElementwiseDescriptor { struct
194 ElementwiseDescriptor&& descriptor);
196 ElementwiseDescriptor&& descriptor,
234 ElementwiseDescriptor&& descriptor);
238 ElementwiseDescriptor&& descriptor,
|
| D | gpu_operation.cc | 490 ElementwiseDescriptor&& descriptor) { in CreateGpuOperation()
496 ElementwiseDescriptor&& descriptor, in CreateGpuOperation()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | depthwise_conv_plus_1x1_conv.cc | 67 void AddElementwiseNode(ElementwiseDescriptor&& op_desc);
473 void ThinPointwiseFuser::AddElementwiseNode(ElementwiseDescriptor&& op_desc) { in AddElementwiseNode()
491 ElementwiseDescriptor op_desc = CreateReLU(attr, op_def_.precision); in AddReluNode()
496 ElementwiseDescriptor op_desc = CreatePReLU(attr, op_def_.dst_tensors[0]); in AddPreluNode()
|