| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/sve2/ |
| H A D | impl.h | 128 const ITensor *broadcast_tensor = is_broadcast_input_2 ? in2 : in1; in elementwise_arithmetic_quantized_op() local
259 const ITensor *broadcast_tensor = is_broadcast_input_2 ? in2 : in1; in elementwise_comparison_quantized_op() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | broadcast_util.cpp | 21 void free_broadcast_tensor(const Tensor& broadcast_tensor) { in free_broadcast_tensor()
140 Tensor broadcast_tensor( in broadcast_tensor() function
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/sve/ |
| H A D | impl.cpp | 59 const ITensor *broadcast_tensor = is_broadcast_input_2 ? in2 : in1; in elementwise_arithmetic_op() local
165 const ITensor *broadcast_tensor = is_broadcast_input_2 ? in2 : in1; in elementwise_comparison_op() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/add/generic/neon/ |
| H A D | impl.cpp | 57 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_same_neon() local
412 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_sub_qasymm8_neon() local
595 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_sub_qasymm8_signed_neon() local
|
| H A D | qsymm16.cpp | 65 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_qsymm16_neon() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/neon/ |
| H A D | impl.h | 107 const ITensor *broadcast_tensor = is_broadcast_input_2 ? in2 : in1; in elementwise_op() local
781 const ITensor *broadcast_tensor = is_broadcast_input_2 ? in2 : in1; in elementwise_op_quantized() local
888 const ITensor *broadcast_tensor = is_broadcast_input_2 ? in2 : in1; in elementwise_comp_quantized_signed() local
995 const ITensor *broadcast_tensor = is_broadcast_input_2 ? in2 : in1; in elementwise_op_quantized_signed() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuMulKernel.cpp | 178 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src2 : src1; in mul_saturate_quantized_8() local
1066 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src2 : src1; in mul_S32_S32_S32() local
1220 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src2 : src1; in mul_F32_F32_F32() local
1320 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src2 : src1; in c_mul_F32_F32_F32_n() local
1452 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src2 : src1; in mul_F16_F16_F16() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/sub/neon/ |
| H A D | list.h | 76 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in sub_same_neon() local
|
| H A D | qsymm16.cpp | 65 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in sub_qsymm16_neon() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/add/generic/sve/ |
| H A D | impl.cpp | 61 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_same_sve() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/add/generic/sve2/ |
| H A D | qsymm16.cpp | 67 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_qsymm16_sve2() local
|
| H A D | qasymm8_signed.cpp | 65 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_qasymm8_signed_sve2() local
|
| H A D | qasymm8.cpp | 66 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_qasymm8_sve2() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
| H A D | NELogicalKernel.cpp | 233 const ITensor *broadcast_tensor = is_broadcast_input_1 ? src1 : src0; in run_binary() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | dynamic_ragged_shape.py | 1979 def broadcast_tensor(self, tensor): member in _LayerBroadcaster
|