Searched defs:OneHot (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | one_hot_op.h | 63 struct OneHot { struct 75 struct OneHot<CPUDevice, T, TI> { argument
|
D | one_hot_op_test.cc | 25 static Graph* OneHot(int batch_size, int num_classes, int axis) { in OneHot() function
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_helpers.cc | 82 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 | 1392 class OneHot : public BuiltinOperator<OneHotOperator, ::tflite::OneHotOptions, class
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 2449 TEST_F(OpTest, OneHot) { in TEST_F() argument
|
/external/tensorflow/tensorflow/go/op/ |
D | wrappers.go | 948 func OneHot(scope *Scope, indices tf.Output, depth tf.Output, on_value tf.Output, off_value tf.Outp… func
|