Searched refs:CreateElementwiseTwoInput (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
H A D | elementwise.cc | 248 ElementwiseDescriptor CreateElementwiseTwoInput( in CreateElementwiseTwoInput() function 274 ElementwiseDescriptor CreateElementwiseTwoInput( in CreateElementwiseTwoInput() function 319 return CreateElementwiseTwoInput(gpu_info, definition, op_type, in CreateElementwiseDesc() 323 return CreateElementwiseTwoInput(gpu_info, definition, op_type, *hwc_tensor, in CreateElementwiseDesc() 349 GPUOperation CreateElementwiseTwoInput(const OperationDef& definition, in CreateElementwiseTwoInput() function
|
H A D | elementwise_test_util.cc | 518 GPUOperation operation = CreateElementwiseTwoInput( in SubTest() 548 GPUOperation operation = CreateElementwiseTwoInput( in SquaredDiffTest() 578 GPUOperation operation = CreateElementwiseTwoInput( in DivTest() 608 GPUOperation operation = CreateElementwiseTwoInput( in PowTest() 638 GPUOperation operation = CreateElementwiseTwoInput( in AddTest() 668 GPUOperation operation = CreateElementwiseTwoInput( in MaximumTest() 824 GPUOperation operation = CreateElementwiseTwoInput( in MinimumTest() 883 GPUOperation operation = CreateElementwiseTwoInput( in MulTest() 913 GPUOperation operation = CreateElementwiseTwoInput( in MulBroadcastHWTest() 943 GPUOperation operation = CreateElementwiseTwoInput( in MulBroadcastChannelsTest() [all …]
|
H A D | elementwise.h | 61 GPUOperation CreateElementwiseTwoInput(const OperationDef& definition,
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
H A D | gpu_model_test_util.cc | 241 GPUOperation mul0_operation = CreateElementwiseTwoInput( in TestLinkingConvolution2InputMul2InputMul() 249 GPUOperation mul1_operation = CreateElementwiseTwoInput( in TestLinkingConvolution2InputMul2InputMul() 380 GPUOperation mul0_operation = CreateElementwiseTwoInput( in TestLinkingConvolution2InputBroadcastMul2InputMul() 388 GPUOperation mul1_operation = CreateElementwiseTwoInput( in TestLinkingConvolution2InputBroadcastMul2InputMul() 519 GPUOperation mul0_operation = CreateElementwiseTwoInput( in TestLinkingConvolution2InputMul2InputBroadcastMul() 527 GPUOperation mul1_operation = CreateElementwiseTwoInput( in TestLinkingConvolution2InputMul2InputBroadcastMul() 665 GPUOperation mul0_operation = CreateElementwiseTwoInput( in TestLinkingConvolution2InputMul2InputMulCos() 673 GPUOperation mul1_operation = CreateElementwiseTwoInput( in TestLinkingConvolution2InputMul2InputMulCos() 799 GPUOperation sub_operation = CreateElementwiseTwoInput( in TestLinkingConvolutionFirstTanh2InputDiff() 917 GPUOperation sub_operation = CreateElementwiseTwoInput( in TestLinkingConvolutionSecondTanh2InputDiff() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
H A D | operation_selector.cc | 776 operation = CreateElementwiseTwoInput(op_def, op_type, in GPUOperationFromNodePart0()
|