Searched defs:OneHot (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-088.md | 3 ### CVE Number
6 ### Impact
22 ### Patches
30 ### For more information
36 ### Attribution
|
| /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/lite/delegates/gpu/cl/kernels/ |
| D | one_hot_test.cc | 28 TEST_F(OpenCLOperationTest, OneHot) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 90 Status XlaHelpers::OneHot(xla::XlaBuilder* builder, int64_t depth, int axis, in OneHot() function in tensorflow::XlaHelpers
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 1043 void OneHot(const int32_t input_rank, in OneHot() function
|
| D | onedevice_train_palm_detection_filter_inference.cc | 1044 void OneHot(const int32_t input_rank, in OneHot() function
|
| D | onedevice_train_palm_detection_filter_inference_v2.cc | 1043 void OneHot(const int32_t input_rank, in OneHot() function
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1391 class OneHot : public BuiltinOperator<OneHotOperator, ::tflite::OneHotOptions, class
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3552 TEST_F(OpTest, OneHot) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 28226 func OneHot(scope *Scope, indices tf.Output, depth tf.Output, on_value tf.Output, off_value tf.Outp… func
|