| /external/tensorflow/tensorflow/core/kernels/ |
| D | one_hot_op.cc | 116 functor::OneHot<Device, T, TI>::Compute(ctx->eigen_device<Device>(), in Compute()
149 void OneHot<GPUDevice, T, TI>::Compute( \
154 extern template struct OneHot<GPUDevice, T, TI>;
|
| D | one_hot_op_gpu.cu.cc | 32 template struct functor::OneHot<GPUDevice, T, TI>;
|
| D | one_hot_op.h | 57 struct OneHot { struct
|
| /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/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 | 212 XlaHelpers::OneHot( in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | one_hot_fuzz.cc | 31 (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/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 | 1734 class OneHot : public BuiltinOperator<OneHotOperator, ::tflite::OneHotOptions, class
2458 ops.push_back(MakeUnique<OneHot>(::tflite::BuiltinOperator_ONE_HOT, in BuildOperatorList()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 363 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/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 19155 name: "OneHot"
19189 name: "OneHot"
19236 name: "OneHot"
|
| D | ops_history.v1.pbtxt | 42435 name: "OneHot"
|
| D | ops_history.v2.pbtxt | 41912 name: "OneHot"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2172 name: "OneHot"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2172 name: "OneHot"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 6730 func OneHot(scope *Scope, indices tf.Output, depth tf.Output, on_value tf.Output, off_value tf.Outp… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 21096 name: "OneHot"
|