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

  /external/tensorflow/tensorflow/core/kernels/
argmax_op.h 29 struct ArgMax {
34 output.device(d) = input.argmax(dimension).template cast<Tout>(); \
  /external/tensorflow/tensorflow/lite/kernels/internal/reference/
legacy_reference_ops.h     [all...]
  /external/tensorflow/tensorflow/lite/toco/tflite/
operator.cc     [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 1545 milliseconds