HomeSort by relevance Sort by last modified time
    Searched refs:softmax_size (Results 1 - 3 of 3) sorted by null

  /external/libtextclassifier/lang_id/common/
embedding-network-params.h 145 return softmax_size() == 1;
258 virtual int softmax_size() const = 0;
  /external/libtextclassifier/lang_id/common/flatbuffers/
embedding-network-params-from-flatbuffer.h 129 int softmax_size() const override { return (SafeGetNumLayers() > 0) ? 1 : 0; }
151 int softmax_bias_size() const override { return softmax_size(); }
  /external/tensorflow/tensorflow/lite/toco/
export_tensorflow.cc 819 const string softmax_size = src_op.outputs[0] + "\/softmax_insert_size"; local
860 const string softmax_size = src_op.outputs[0] + "\/log_softmax_insert_size"; local
    [all...]

Completed in 108 milliseconds