| /external/tensorflow/tensorflow/core/kernels/ |
| D | one_hot_op.cc | 117 functor::OneHot<Device, T, TI>::Compute(ctx->eigen_device<Device>(), in Compute()
151 void OneHot<GPUDevice, T, TI>::Compute( \
156 extern template struct OneHot<GPUDevice, T, TI>;
|
| D | one_hot_op_test.cc | 25 static Graph* OneHot(int batch_size, int num_classes, int axis) { in OneHot() function
61 test::Benchmark(#DEVICE, OneHot(BATCH, CLASS, AXIS), \
|
| D | one_hot_op.h | 63 struct OneHot { struct
75 struct OneHot<CPUDevice, T, TI> { argument
|
| D | one_hot_op_gpu.cu.cc | 33 template struct functor::OneHot<GPUDevice, T, TI>;
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_OneHot.pbtxt | 2 graph_op_name: "OneHot"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_OneHot.pbtxt | 2 graph_op_name: "OneHot"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | OneHot.pbtxt | 2 name: "OneHot"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | OneHot.pbtxt | 2 name: "OneHot"
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | ops_defs.py | 55 rdims = tf.raw_ops.OneHot(
105 rdims = tf.raw_ops.OneHot(
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | one_hot_op.cc | 67 ctx, XlaHelpers::OneHot(ctx->builder(), depth, axis, input_type(0), in Compile()
|
| D | softmax_op.cc | 231 XlaHelpers::OneHot( in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | one_hot_fuzz.cc | 34 (void)tensorflow::ops::OneHot(scope.WithOpName("output"), input, depth, on, in BuildGraph()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.h | 62 static Status OneHot(xla::XlaBuilder* builder, int64 depth, int axis,
|
| D | xla_helpers.cc | 82 Status XlaHelpers::OneHot(xla::XlaBuilder* builder, int64 depth, int axis, in OneHot() function in tensorflow::XlaHelpers
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_OneHot.pbtxt | 2 graph_op_name: "OneHot"
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen_test.py | 94 x = array_ops.OneHot(
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 294 * `raw_ops.OneHot`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 153 `OneHot` | `TI={int32,int64}`<br>`T={bool,complex64,double,float,int32…
|
| D | cpu_supported_ops.md | 153 `OneHot` | `TI={int32,int64}`<br>`T={bool,complex64,double,float,int32…
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1392 class OneHot : public BuiltinOperator<OneHotOperator, ::tflite::OneHotOptions, class
1957 ops.push_back(MakeUnique<OneHot>(::tflite::BuiltinOperator_ONE_HOT, in BuildOperatorList()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf-ops.mlir | 2878 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
2887 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
2895 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
2903 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
2911 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
2919 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -2 : i64} : (tensor<3xi32>,…
|
| D | lower_tf.mlir | 454 …// CHECK-DAG: %[[LABELS:.*]] = "tf.OneHot"(%[[SPARSE_LABELS]], %[[DEPTH]], %[[ONE]], %[[ZERO]]) {a…
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf.mlir | 1380 func @OneHot(%arg0: tensor<3xi32>, %arg1: tensor<i32>, %arg2: tensor<f32>, %arg3: tensor<f32>) -> t…
1381 …%0 = "tf.OneHot"(%arg0, %arg1, %arg2, %arg3) {axis = -1 : i64} : (tensor<3xi32>, tensor<i32>, tens…
1384 // CHECK-LABEL: OneHot
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 382 HANDLE_OPERATORTYPENAME_CASE(OneHot) in OperatorTypeName()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2449 TEST_F(OpTest, OneHot) { in TEST_F() argument
|