| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_one_hot_op_test.py | 85 on_value=None, argument
119 on_value=None, argument
153 on_value=None, argument
|
| H A D | ragged_array_ops.py | 547 on_value=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | one_hot.cc | 55 const TfLiteTensor* on_value; member
80 const T on_value = *GetTensorData<T>(op_context.on_value); in OneHotComputeImpl() local
|
| H A D | one_hot_test.cc | 36 TensorType dtype, int axis = -1, T on_value = 1, in OneHotOpModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | one_hot_op.h | 41 const typename TTypes<T>::ConstScalar& on_value, in OneGenerator()
|
| H A D | one_hot_op_test.cc | 41 Tensor on_value(DT_FLOAT, TensorShape({})); in OneHot() local
|
| H A D | one_hot_op.cc | 54 const Tensor& on_value = ctx->input(2); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_helpers.cc | 92 const xla::XlaOp& indices, const xla::XlaOp& on_value, in OneHot()
|
| /aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| H A 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()
|
| H A D | onedevice_train_palm_detection_filter_inference.cc | 1048 const T* __restrict on_value, in OneHot()
1075 const T* __restrict on_value, in OneHotLastDim()
|
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | operations.h | 621 float on_value = 1; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 163 static TensorType InferOneHotOpType(Value indices, Value depth, Value on_value, in InferOneHotOpType()
186 Value depth, Value on_value, Value off_value, in build()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | array_ops.py | 4306 on_value=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_common.cc | 3834 Value on_value, Value off_value, in convertOneHotOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 5334 Value on_value = rewriter.create<BroadcastOp>( in matchAndRewrite() local
|