Searched refs:softmax_size (Results 1 – 3 of 3) sorted by relevance
145 return softmax_size() == 1; in HasSoftmax()258 virtual int softmax_size() const = 0;
129 int softmax_size() const override { return (SafeGetNumLayers() > 0) ? 1 : 0; } in softmax_size() function151 int softmax_bias_size() const override { return softmax_size(); } in softmax_bias_size()
825 const std::string softmax_size = src_op.outputs[0] + "/softmax_insert_size"; in ConvertSoftmaxOperator() local832 *reshape_op->add_input() = softmax_size; in ConvertSoftmaxOperator()842 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph); in ConvertSoftmaxOperator()866 const std::string softmax_size = in ConvertLogSoftmaxOperator() local874 *reshape_op->add_input() = softmax_size; in ConvertLogSoftmaxOperator()884 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph); in ConvertLogSoftmaxOperator()