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

  /external/tensorflow/tensorflow/core/kernels/
eigen_softmax_test.cc 56 Tensor<float, 2> result = SoftMax(input, beta);
eigen_softmax.h 23 /** SoftMax
26 * \brief Applies a softmax
87 SoftMax(const Input& input, const float beta) {
  /frameworks/ml/nn/runtime/test/specs/V1_0/
mobilenet_quantized.mod.py     [all...]
  /frameworks/ml/nn/tools/test_generator/
test_generator.py 527 def SoftMax(self, input, beta):
530 op = Operation("SOFTMAX", ins, outs)

Completed in 2079 milliseconds