Searched defs:on_value (Results 1 – 11 of 11) sorted by relevance
| /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/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/lite/kernels/ |
| D | one_hot.cc | 55 const TfLiteTensor* on_value; member
75 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/compiler/tf2xla/ |
| D | xla_helpers.cc | 84 const xla::XlaOp& indices, const xla::XlaOp& on_value, in OneHot()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 146 static TensorType InferOneHotOpType(Value indices, Value depth, Value on_value, in InferOneHotOpType()
169 Value depth, Value on_value, Value off_value, in build()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 4180 on_value=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4530 Value on_value = rewriter.create<BroadcastOp>( in matchAndRewrite() local
|