Searched refs:softmax_input (Results 1 – 1 of 1) sorted by relevance
811 string softmax_input; in ConvertSoftmaxOperator() local814 softmax_input = src_op.inputs[0]; in ConvertSoftmaxOperator()820 softmax_input = reshape_output; in ConvertSoftmaxOperator()842 *softmax_op->add_input() = softmax_input; in ConvertSoftmaxOperator()851 string softmax_input; in ConvertLogSoftmaxOperator() local854 softmax_input = src_op.inputs[0]; in ConvertLogSoftmaxOperator()861 softmax_input = reshape_output; in ConvertLogSoftmaxOperator()883 *log_softmax_op->add_input() = softmax_input; in ConvertLogSoftmaxOperator()