| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | one_hot_op_test.py | 57 on_value = np.asarray(1.0, dtype=dtype)
69 on_value=on_value,
78 on_value=on_value,
130 on_value = np.asarray(1.0, dtype=dtype)
143 on_value=on_value,
152 on_value=on_value,
182 on_value = np.asarray(1.0, dtype=dtype)
194 on_value=on_value,
203 on_value=on_value,
213 on_value = np.asarray(1.0, dtype=dtype)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_one_hot_op_test.py | 54 on_value=8,
85 on_value=None, argument
96 on_value=on_value,
119 on_value=None, argument
132 on_value=on_value,
147 dict(indices_shape=[5, 2, 7], depth=3, on_value=True, off_value=False),
153 on_value=None, argument
176 on_value=on_value,
188 on_value=on_value,
|
| D | ragged_array_ops.py | 547 on_value=None, argument
562 [indices, depth, on_value, off_value, axis]):
566 array_ops.one_hot(indices.flat_values, depth, on_value, off_value, axis,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | one_hot_op.h | 41 const typename TTypes<T>::ConstScalar& on_value, in OneGenerator() argument
43 : indices_(indices), on_value_(on_value), off_value_(off_value) {} in OneGenerator()
66 const typename TTypes<T>::ConstScalar& on_value, in Compute()
69 generator::OneGenerator<T, TI> generator(indices, on_value, off_value); in Compute()
78 const typename TTypes<T>::ConstScalar& on_value,
100 (*output)(i, depth, 0) = on_value();
112 (*output)(d0, depth, d1) = on_value();
|
| D | one_hot_op.cc | 54 const Tensor& on_value = ctx->input(2); in Compute() local
69 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(on_value.shape()), in Compute()
71 on_value.shape().DebugString())); in Compute()
93 auto on_value_t = on_value.scalar<T>(); in Compute()
153 const typename TTypes<T>::ConstScalar& on_value, \
|
| D | one_hot_op_test.cc | 41 Tensor on_value(DT_FLOAT, TensorShape({})); in OneHot() local
42 on_value.scalar<float>()() = 1.0f; in OneHot()
51 test::graph::Constant(g, on_value), in OneHot()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | one_hot.cc | 41 on_value = GetInput(context, node, kOnValueTensor); in OneHotContext()
50 dtype = on_value->type; in OneHotContext()
55 const TfLiteTensor* on_value; member
80 const T on_value = *GetTensorData<T>(op_context.on_value); in OneHotComputeImpl() local
95 ? on_value in OneHotComputeImpl()
154 TF_LITE_ENSURE_EQ(context, NumElements(op_context.on_value), 1); in Prepare()
156 TF_LITE_ENSURE_TYPES_EQ(context, op_context.on_value->type, op_context.dtype); in Prepare()
|
| D | one_hot_test.cc | 36 TensorType dtype, int axis = -1, T on_value = 1, in OneHotOpModel() argument
48 PopulateTensor<T>(on, {on_value}); in OneHotOpModel()
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | one_hot.py | 46 on_value = tf.compat.v1.placeholder(
53 on_value=on_value,
57 return [indices, depth, on_value, off_value], [out]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_OneHot.pbtxt | 16 name: "on_value"
41 The locations represented by indices in `indices` take value `on_value`,
72 on_value = 5.0
90 on_value = 0.0
112 on_value = 1.0
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | one_hot.cc | 66 op.args_.AddFloat("on_value", attr.on_value); in CreateOneHot()
69 op.args_.AddHalf("on_value", half(attr.on_value)); in CreateOneHot()
|
| D | one_hot_test_util.cc | 78 expected[i * 10 + i] = attr.on_value; in OneHotBatchTest()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | OneHot.pbtxt | 12 name: "on_value"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | OneHot.pbtxt | 12 name: "on_value"
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | ops_defs.py | 52 indices=i, depth=len(shape), on_value=True, off_value=False, axis=-1)
102 indices=i, depth=len(shape), on_value=True, off_value=False, axis=-1)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 92 const xla::XlaOp& indices, const xla::XlaOp& on_value, in OneHot() argument
109 xla::Broadcast(on_value, output_shape.dim_sizes()), in OneHot()
|
| D | xla_helpers.h | 71 const xla::XlaOp& indices, const xla::XlaOp& on_value,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | nary_ops_test.py | 130 on_value=np.float32(7), off_value=np.float32(3))
139 on_value=np.int32(2), off_value=np.int32(1),
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 4306 on_value=None, argument
4420 [indices, depth, on_value, off_value, axis, dtype]) as name:
4421 on_exists = on_value is not None
4425 on_value = ops.convert_to_tensor(on_value, dtype_hint=dtype)
4429 on_dtype = on_value.dtype.base_dtype if on_exists else None
4450 on_value = ops.convert_to_tensor(1, dtype, name="on_value")
4461 return gen_array_ops.one_hot(indices, depth, on_value, off_value, axis,
|
| D | ctc_ops.py | 580 on_value=0.0,
621 on_value=0.0,
1277 on_value=0.0,
|
| /external/rust/crates/xml-rs/src/reader/parser/ |
| D | mod.rs | 443 fn read_attribute_value<F>(&mut self, t: Token, on_value: F) -> Option<Result> in read_attribute_value()
456 on_value(self, value) in read_attribute_value()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-003.md | 28 data = tf.one_hot(indices=[62,50],depth=136,on_value=np.int32(237),off_value=158,axis=856,dtype=20)
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen_test.py | 89 indices=[0, 2, -1, x1], depth=y1, on_value=True, off_value=False)
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | softmax_spmd_expander.cc | 308 TF_ASSIGN_OR_RETURN(mlir::Value on_value, in ComputeOneHot()
314 .create<mlir::TF::OneHotOp>(input.getLoc(), indices, depth, on_value, in ComputeOneHot()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 269 Value on_value, Value off_value,
|