Home
last modified time | relevance | path

Searched defs:OneHot (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Done_hot_op.h57 struct OneHot { struct
58 EIGEN_ALWAYS_INLINE static void Compute( in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc84 Status XlaHelpers::OneHot(xla::XlaBuilder* builder, int64 depth, int axis, in OneHot() function in tensorflow::XlaHelpers
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1734 class OneHot : public BuiltinOperator<OneHotOperator, ::tflite::OneHotOptions, class
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2448 TEST_F(OpTest, OneHot) { in TEST_F() argument
/external/tensorflow/tensorflow/go/op/
Dwrappers.go6730 func OneHot(scope *Scope, indices tf.Output, depth tf.Output, on_value tf.Output, off_value tf.Outp… func