| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_one_hot_op_test.py | 88 on_value=None, argument
121 on_value=None, argument
155 on_value=None, argument
|
| D | ragged_array_ops.py | 519 on_value=None, argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | one_hot.cc | 55 const TfLiteTensor* on_value; member
80 const T on_value = *GetTensorData<T>(op_context.on_value); in OneHotComputeImpl() local
|
| D | one_hot_test.cc | 36 TensorType dtype, int axis = -1, T on_value = 1, in OneHotOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | one_hot_op.h | 41 const typename TTypes<T>::ConstScalar& on_value, in OneGenerator()
|
| D | one_hot_op_test.cc | 41 Tensor on_value(DT_FLOAT, TensorShape({})); in OneHot() local
|
| D | one_hot_op.cc | 54 const Tensor& on_value = ctx->input(2); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 89 const xla::XlaOp& indices, const xla::XlaOp& on_value, in OneHot()
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference.cc | 1048 const T* __restrict on_value, in OneHot()
1075 const T* __restrict on_value, in OneHotLastDim()
|
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 1047 const T* __restrict on_value, in OneHot()
1074 const T* __restrict on_value, in OneHotLastDim()
|
| D | onedevice_train_palm_detection_filter_inference_v2.cc | 1047 const T* __restrict on_value, in OneHot()
1074 const T* __restrict on_value, in OneHotLastDim()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 147 static TensorType InferOneHotOpType(Value indices, Value depth, Value on_value, in InferOneHotOpType()
170 Value depth, Value on_value, Value off_value, in build()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 4230 on_value=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 3545 Value on_value, Value off_value, in convertOneHotOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5434 Value on_value = rewriter.create<BroadcastOp>( in matchAndRewrite() local
|