Searched refs:softmax_input (Results 1 – 2 of 2) sorted by relevance
47 const std::string softmax_input = "softmax_input"; in TEST() local49 softmax->inputs.push_back(softmax_input); in TEST()51 array_map[softmax_input] = std::unique_ptr<Array>(new Array); in TEST()52 array_map[softmax_input]->data_type = ArrayDataType::kFloat; in TEST()
816 std::string softmax_input; in ConvertSoftmaxOperator() local819 softmax_input = src_op.inputs[0]; in ConvertSoftmaxOperator()826 softmax_input = reshape_output; in ConvertSoftmaxOperator()848 *softmax_op->add_input() = softmax_input; in ConvertSoftmaxOperator()857 std::string softmax_input; in ConvertLogSoftmaxOperator() local860 softmax_input = src_op.inputs[0]; in ConvertLogSoftmaxOperator()868 softmax_input = reshape_output; in ConvertLogSoftmaxOperator()890 *log_softmax_op->add_input() = softmax_input; in ConvertLogSoftmaxOperator()