Searched defs:softmax_op (Results 1 – 5 of 5) sorted by relevance
34 xnn_operator_t softmax_op = NULL; in xnn_create_softmax_nc_qu8() local141 xnn_operator_t softmax_op, in xnn_setup_softmax_nc_qu8()197 xnn_operator_t softmax_op = NULL; in create_softmax_nc_floating_point() local297 xnn_operator_t softmax_op, in setup_softmax_nc_floating_point()367 xnn_operator_t softmax_op, in xnn_setup_softmax_nc_f16()399 xnn_operator_t softmax_op, in xnn_setup_softmax_nc_f32()
143 xnn_operator_t softmax_op = nullptr; in TestF16() local207 xnn_operator_t softmax_op = nullptr; in TestF32() local276 xnn_operator_t softmax_op = nullptr; in TestQU8() local
46 xnn_operator_t softmax_op = nullptr; in xnnpack_softmax_qu8() local115 xnn_operator_t softmax_op = nullptr; in xnnpack_softmax_f32() local
99 std::unique_ptr<pytorch_qnnp_operator, QnnpackOperatorDeleter> softmax_op( in qsoftmax_qnnpack() local
850 tensorflow::NodeDef* softmax_op = tensorflow_graph->add_node(); in ConvertSoftmaxOperator() local