Home
last modified time | relevance | path

Searched refs:softmax_size (Results 1 – 3 of 3) sorted by relevance

/external/libtextclassifier/lang_id/common/
Dembedding-network-params.h145 return softmax_size() == 1; in HasSoftmax()
258 virtual int softmax_size() const = 0;
/external/libtextclassifier/lang_id/common/flatbuffers/
Dembedding-network-params-from-flatbuffer.h129 int softmax_size() const override { return (SafeGetNumLayers() > 0) ? 1 : 0; } in softmax_size() function
151 int softmax_bias_size() const override { return softmax_size(); } in softmax_bias_size()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc819 const string softmax_size = src_op.outputs[0] + "/softmax_insert_size"; in ConvertSoftmaxOperator() local
826 *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() local
867 *reshape_op->add_input() = softmax_size; in ConvertLogSoftmaxOperator()
877 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph); in ConvertLogSoftmaxOperator()