| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_one_hot_op_test.py | 86 off_value=None, argument
120 off_value=None, argument
154 off_value=None, argument
|
| H A D | ragged_array_ops.py | 548 off_value=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | one_hot.cc | 56 const TfLiteTensor* off_value; member
81 const T off_value = *GetTensorData<T>(op_context.off_value); in OneHotComputeImpl() local
|
| H A D | one_hot_test.cc | 37 T off_value = 0, TensorType indices_type = TensorType_INT32) { in OneHotOpModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | one_hot_op.h | 42 const typename TTypes<T>::ConstScalar& off_value) in OneGenerator()
|
| H A D | one_hot_op_test.cc | 44 Tensor off_value(DT_FLOAT, TensorShape({})); in OneHot() local
|
| H A D | one_hot_op.cc | 55 const Tensor& off_value = ctx->input(3); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_helpers.cc | 93 const xla::XlaOp& off_value, xla::XlaOp* one_hot) { 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 | 1048 const T* __restrict off_value, in OneHot()
1075 const T* __restrict off_value, in OneHotLastDim()
|
| H A D | onedevice_train_palm_detection_filter_inference.cc | 1049 const T* __restrict off_value, in OneHot()
1076 const T* __restrict off_value, in OneHotLastDim()
|
| H A D | onedevice_train_palm_detection_filter_inference_v2.cc | 1048 const T* __restrict off_value, in OneHot()
1075 const T* __restrict off_value, in OneHotLastDim()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | operations.h | 622 float off_value = 0; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 164 Value off_value, IntegerAttr axis) { in InferOneHotOpType()
186 Value depth, Value on_value, Value off_value, in build()
|
| /aosp_15_r20/art/compiler/optimizing/ |
| H A D | induction_var_range.cc | 407 int64_t off_value = 0; in IsUnitStride() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | array_ops.py | 4307 off_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 | 5337 Value off_value = rewriter.create<BroadcastOp>( in matchAndRewrite() local
|