Home
last modified time | relevance | path

Searched defs:output_arg (Results 1 – 25 of 107) sorted by relevance

12345

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DScanKernels.cpp29 TensorArg output_arg{ values, "output", 1 }; in cummax_helper_cuda() local
46 TensorArg output_arg{ values, "output", 1 }; in cummin_helper_cuda() local
74 TensorArg output_arg{ result, "output", 1 }; in _logcumsumexp_out_cuda() local
H A DMaxUnpooling.cu131 TensorArg output_arg{output, "output", 1}, self_arg{self_, "self_", 2}, in max_unpooling2d_forward_out_cuda() local
309 TensorArg output_arg{output, "output", 1}, self_arg{self_, "self_", 2}, in max_unpooling3d_forward_out_cuda() local
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/gen/cpp/renderers/
H A Dop_implementation_renderer.cc63 ArgView output_arg = op_.OnlyOutput(); in RenderExecutionListOp() local
70 ArgView output_arg = op_.OnlyOutput(); in RenderExecutionSingleOutput() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
H A Dgemm_hybrid_indirect.hpp78 …unsigned int kern_k, const Tro *b_ptr, size_t, IndirectOutputArg<Tr> output_arg, const Tr *bias_pt… in run()
124 …unsigned int kern_k, const Tro *b_ptr, size_t b_stride, IndirectOutputArg<Tr> output_arg, const Tr… in run()
172 …unsigned int kern_k, const Tro *b_ptr, size_t, IndirectOutputArg<Tr> output_arg, const Tr *, Activ… in run()
189 …unsigned int kern_k, const Tro *b_ptr, size_t, IndirectOutputArg<Tr> output_arg, const Tr *, Activ… in run()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dfunctions.cc62 for (const OutputArgInstantiation& output_arg : output_args_) { in GrapplerFunctionItem() local
452 for (const OutputArgInstantiation& output_arg : item.outputs()) { in Initialize() local
553 for (const OutputArgInstantiation& output_arg : item.outputs()) { in MakeFunctionDef() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DParamUtils.cpp39 TensorArg grad_arg{grad_, "grad", 1}, output_arg{output_, "output", 2}; in softmax_backward_sparse_input_preprocessing() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8u32_dot_6x4VL/
H A Dgeneric.cpp35 size_t M, size_t N, const uint8_t *B_ptr, IndirectOutputArg<uint32_t> output_arg, in sve_hybrid_u8u32_dot_6x4VL()
H A Da64fx.cpp35 size_t M, size_t N, const uint8_t *B_ptr, IndirectOutputArg<uint32_t> output_arg, in sve_hybrid_u8u32_dot_6x4VL_a64fx()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8qa_dot_4x16/
H A Da55.cpp36 size_t M, size_t N, const int8_t *B_ptr, IndirectOutputArg<int8_t> output_arg, in a64_hybrid_s8qa_dot_4x16_a55()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8u32_mmla_6x4VL/
H A Dgeneric.cpp35 size_t M, size_t N, const uint8_t *B_ptr, IndirectOutputArg<uint32_t> output_arg, in sve_hybrid_u8u32_mmla_6x4VL()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8qa_dot_4x4VL/
H A Dgeneric.cpp36 size_t M, size_t N, const uint8_t *B_ptr, IndirectOutputArg<uint8_t> output_arg, in sve_hybrid_u8qa_dot_4x4VL()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_s8qa_dot_4x4VL/
H A Dgeneric.cpp36 size_t M, size_t N, const int8_t *B_ptr, IndirectOutputArg<int8_t> output_arg, in sve_hybrid_s8qa_dot_4x4VL()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8s32_dot_6x16/
H A Da55.cpp35 size_t M, size_t N, const int8_t *B_ptr, IndirectOutputArg<int32_t> output_arg, in a64_hybrid_s8s32_dot_6x16_a55()
H A Dgeneric.cpp35 size_t M, size_t N, const int8_t *B_ptr, IndirectOutputArg<int32_t> output_arg, in a64_hybrid_s8s32_dot_6x16()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8s32_mmla_6x16/
H A Dgeneric.cpp35 size_t M, size_t N, const int8_t *B_ptr, IndirectOutputArg<int32_t> output_arg, in a64_hybrid_s8s32_mmla_6x16()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8qa_mmla_4x16/
H A Dgeneric.cpp36 size_t M, size_t N, const uint8_t *B_ptr, IndirectOutputArg<uint8_t> output_arg, in a64_hybrid_u8qa_mmla_4x16()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_s8qa_mmla_4x4VL/
H A Dgeneric.cpp36 size_t M, size_t N, const int8_t *B_ptr, IndirectOutputArg<int8_t> output_arg, in sve_hybrid_s8qa_mmla_4x4VL()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_s8s32_mmla_6x4VL/
H A Dgeneric.cpp35 size_t M, size_t N, const int8_t *B_ptr, IndirectOutputArg<int32_t> output_arg, in sve_hybrid_s8s32_mmla_6x4VL()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8qa_mmla_4x4VL/
H A Dgeneric.cpp36 size_t M, size_t N, const uint8_t *B_ptr, IndirectOutputArg<uint8_t> output_arg, in sve_hybrid_u8qa_mmla_4x4VL()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8qa_mmla_4x16/
H A Dgeneric.cpp36 size_t M, size_t N, const int8_t *B_ptr, IndirectOutputArg<int8_t> output_arg, in a64_hybrid_s8qa_mmla_4x16()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_s8s32_dot_6x4VL/
H A Da64fx.cpp35 size_t M, size_t N, const int8_t *B_ptr, IndirectOutputArg<int32_t> output_arg, in sve_hybrid_s8s32_dot_6x4VL_a64fx()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8qa_dot_4x16/
H A Dgeneric.cpp36 size_t M, size_t N, const uint8_t *B_ptr, IndirectOutputArg<uint8_t> output_arg, in a64_hybrid_u8qa_dot_4x16()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8u32_dot_6x16/
H A Da55.cpp35 size_t M, size_t N, const uint8_t *B_ptr, IndirectOutputArg<uint32_t> output_arg, in a64_hybrid_u8u32_dot_6x16_a55()
H A Dgeneric.cpp35 size_t M, size_t N, const uint8_t *B_ptr, IndirectOutputArg<uint32_t> output_arg, in a64_hybrid_u8u32_dot_6x16()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8u32_mmla_6x16/
H A Dgeneric.cpp35 size_t M, size_t N, const uint8_t *B_ptr, IndirectOutputArg<uint32_t> output_arg, in a64_hybrid_u8u32_mmla_6x16()

12345