Searched defs:off_value (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_one_hot_op_test.py | 89 off_value=None, argument
122 off_value=None, argument
156 off_value=None, argument
|
| D | ragged_array_ops.py | 520 off_value=None, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | one_hot_op.h | 42 const typename TTypes<T>::ConstScalar& off_value) in OneGenerator()
|
| D | one_hot_op_test.cc | 44 Tensor off_value(DT_FLOAT, TensorShape({})); in OneHot() local
|
| D | one_hot_op.cc | 55 const Tensor& off_value = ctx->input(3); in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | one_hot.cc | 56 const TfLiteTensor* off_value; member
76 const T off_value = *GetTensorData<T>(op_context.off_value); in OneHotComputeImpl() local
|
| D | one_hot_test.cc | 37 T off_value = 0, TensorType indices_type = TensorType_INT32) { in OneHotOpModel()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 85 const xla::XlaOp& off_value, xla::XlaOp* one_hot) { in OneHot()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 147 Value off_value, IntegerAttr axis) { in InferOneHotOpType()
169 Value depth, Value on_value, Value off_value, in build()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 4181 off_value=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4533 Value off_value = rewriter.create<BroadcastOp>( in matchAndRewrite() local
|