Searched defs:OneHot (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | one_hot_op.h | 57 struct OneHot { struct 58 EIGEN_ALWAYS_INLINE static void Compute( in Compute()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_helpers.cc | 84 Status XlaHelpers::OneHot(xla::XlaBuilder* builder, int64 depth, int axis, in OneHot() function in tensorflow::XlaHelpers
|
/external/tensorflow/tensorflow/lite/toco/tflite/ |
D | operator.cc | 1734 class OneHot : public BuiltinOperator<OneHotOperator, ::tflite::OneHotOptions, class
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 2448 TEST_F(OpTest, OneHot) { in TEST_F() argument
|
/external/tensorflow/tensorflow/go/op/ |
D | wrappers.go | 6730 func OneHot(scope *Scope, indices tf.Output, depth tf.Output, on_value tf.Output, off_value tf.Outp… func
|