Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Done_hot_op.h63 struct OneHot { struct
75 struct OneHot<CPUDevice, T, TI> { argument
Done_hot_op_test.cc25 static Graph* OneHot(int batch_size, int num_classes, int axis) { in OneHot() function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc82 Status XlaHelpers::OneHot(xla::XlaBuilder* builder, int64 depth, int axis, in OneHot() function in tensorflow::XlaHelpers
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1392 class OneHot : public BuiltinOperator<OneHotOperator, ::tflite::OneHotOptions, class
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2449 TEST_F(OpTest, OneHot) { in TEST_F() argument
/external/tensorflow/tensorflow/go/op/
Dwrappers.go948 func OneHot(scope *Scope, indices tf.Output, depth tf.Output, on_value tf.Output, off_value tf.Outp… func