| /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)).Run(iters); \
|
| 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_v1/ |
| D | OneHot.pbtxt | 2 name: "OneHot"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.h | 61 static Status OneHot(xla::XlaBuilder* builder, int64 depth, int axis,
|
| D | xla_helpers.cc | 84 Status XlaHelpers::OneHot(xla::XlaBuilder* builder, int64 depth, int axis, in OneHot() function in tensorflow::XlaHelpers
|
| /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 | 204 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/core/api_def/base_api/ |
| D | api_def_OneHot.pbtxt | 2 graph_op_name: "OneHot"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf-ops.mlir | 1599 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
1608 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
1616 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
1624 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
1632 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
1640 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -2 : i64} : (tensor<3xi32>,…
|
| D | lower_tf.mlir | 255 …// CHECK-DAG: %[[LABELS:.*]] = "tf.OneHot"(%[[SPARSE_LABELS]], %[[DEPTH]], %[[ONE]], %[[ZERO]]) {a…
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1312 class OneHot : public BuiltinOperator<OneHotOperator, ::tflite::OneHotOptions, class
1880 ops.push_back(MakeUnique<OneHot>(::tflite::BuiltinOperator_ONE_HOT, in BuildOperatorList()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | cpu_supported_ops.md | 153 `OneHot` | `TI={int32,int64}`<br>`T={bool,complex64,double,float,int32…
|
| D | gpu_supported_ops.md | 153 `OneHot` | `TI={int32,int64}`<br>`T={bool,complex64,double,float,int32…
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf.mlir | 1118 func @OneHot(%arg0: tensor<3xi32>, %arg1: tensor<i32>, %arg2: tensor<f32>, %arg3: tensor<f32>) -> t…
1119 …%0 = "tf.OneHot"(%arg0, %arg1, %arg2, %arg3) {axis = -1 : i64} : (tensor<3xi32>, tensor<i32>, tens…
1122 // CHECK-LABEL: OneHot
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 380 HANDLE_OPERATORTYPENAME_CASE(OneHot) in OperatorTypeName()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2448 TEST_F(OpTest, OneHot) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 1277 // OneHot op legalizations.
1289 …%result = "tf.OneHot"(%indices, %depth, %on_value, %off_value) {axis = -1 : i64} : (tensor<3xi32>,…
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 1572 let summary = "OneHot operator";
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 19155 name: "OneHot"
19189 name: "OneHot"
19236 name: "OneHot"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2516 name: "OneHot"
|