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()
819 const string softmax_size = src_op.outputs[0] + "/softmax_insert_size"; in ConvertSoftmaxOperator() local826 *reshape_op->add_input() = softmax_size; in ConvertSoftmaxOperator()836 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph); in ConvertSoftmaxOperator()860 const string softmax_size = src_op.outputs[0] + "/log_softmax_insert_size"; in ConvertLogSoftmaxOperator() local867 *reshape_op->add_input() = softmax_size; in ConvertLogSoftmaxOperator()877 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph); in ConvertLogSoftmaxOperator()