Home
last modified time | relevance | path

Searched defs:broadcast_tensor (Results 1 – 14 of 14) sorted by relevance

/external/ComputeLibrary/src/cpu/kernels/sub/neon/
Dlist.h76 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in sub_same_neon() local
Dqsymm16.cpp65 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in sub_qsymm16_neon() local
/external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/sve2/
Dimpl.h128 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
/external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/sve/
Dimpl.cpp59 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
/external/ComputeLibrary/src/cpu/kernels/add/generic/sve/
Dimpl.cpp61 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_same_sve() local
/external/ComputeLibrary/src/cpu/kernels/add/generic/sve2/
Dqsymm16.cpp67 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_qsymm16_sve2() local
Dqasymm8_signed.cpp65 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_qasymm8_signed_sve2() local
Dqasymm8.cpp66 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_qasymm8_sve2() local
/external/ComputeLibrary/src/cpu/kernels/add/generic/neon/
Dqsymm16.cpp65 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0; in add_qsymm16_neon() local
Dimpl.cpp57 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
/external/ComputeLibrary/src/core/NEON/kernels/
DNELogicalKernel.cpp233 const ITensor *broadcast_tensor = is_broadcast_input_1 ? src1 : src0; in run_binary() local
/external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/neon/
Dimpl.h107 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
/external/ComputeLibrary/src/cpu/kernels/
DCpuMulKernel.cpp178 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
/external/tensorflow/tensorflow/python/ops/ragged/
Ddynamic_ragged_shape.py1979 def broadcast_tensor(self, tensor): member in _LayerBroadcaster