| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | one_hot_op_test.py | 58 on_value = np.asarray(1.0, dtype=dtype)
70 on_value=on_value,
79 on_value=on_value,
128 on_value = np.asarray(1.0, dtype=dtype)
141 on_value=on_value,
150 on_value=on_value,
180 on_value = np.asarray(1.0, dtype=dtype)
192 on_value=on_value,
201 on_value=on_value,
211 on_value = np.asarray(1.0, dtype=dtype)
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | one_hot.cc | 39 on_value = GetInput(context, node, kOnValueTensor); in OneHotContext()
48 dtype = on_value->type; in OneHotContext()
53 const TfLiteTensor* on_value; member
73 const T on_value = *GetTensorData<T>(op_context.on_value); in OneHotComputeImpl() local
88 ? on_value in OneHotComputeImpl()
145 TF_LITE_ENSURE_EQ(context, NumElements(op_context.on_value), 1); in Prepare()
147 TF_LITE_ENSURE_EQ(context, op_context.on_value->type, op_context.dtype); in Prepare()
|
| D | one_hot_test.cc | 33 TensorType dtype, int axis = -1, T on_value = 1, in OneHotOpModel() argument
45 PopulateTensor<T>(on, {on_value}); in OneHotOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | one_hot_op.h | 35 const typename TTypes<T>::ConstScalar& on_value, in OneGenerator() argument
37 : indices_(indices), on_value_(on_value), off_value_(off_value) {} in OneGenerator()
60 const typename TTypes<T>::ConstScalar& on_value, in Compute()
63 generator::OneGenerator<T, TI> generator(indices, on_value, off_value); in Compute()
|
| D | one_hot_op.cc | 53 const Tensor& on_value = ctx->input(2); in Compute() local
68 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(on_value.shape()), in Compute()
70 on_value.shape().DebugString())); in Compute()
92 auto on_value_t = on_value.scalar<T>(); in Compute()
151 const typename TTypes<T>::ConstScalar& on_value, \
|
| /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/compiler/tf2xla/ |
| D | xla_helpers.cc | 86 const xla::XlaOp& indices, const xla::XlaOp& on_value, in OneHot() argument
103 xla::Broadcast(on_value, output_shape.dim_sizes()), in OneHot()
|
| D | xla_helpers.h | 63 const xla::XlaOp& indices, const xla::XlaOp& on_value,
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | params.cc | 43 return param.threshold().on_value(); in ResolveParam()
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | beam_search_decoder.py | 704 on_value=False,
723 on_value=ops.convert_to_tensor(0.0, dtype=dtype),
888 on_value=False,
898 on_value=ops.convert_to_tensor(0.0, dtype=dtype),
985 on_value=np.int64(0),
1294 on_value=ops.convert_to_tensor(0., dtype=probs.dtype),
|
| /external/tensorflow/tensorflow/contrib/kernel_methods/python/ |
| D | losses.py | 130 indices=labels, depth=num_classes, on_value=0.0, off_value=1.0)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | nary_ops_test.py | 133 on_value=np.float32(7), off_value=np.float32(3))
142 on_value=np.int32(2), off_value=np.int32(1),
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 2905 on_value=None, argument
3004 [indices, depth, on_value, off_value, axis,
3006 on_exists = on_value is not None
3009 on_dtype = (ops.convert_to_tensor(on_value).dtype.base_dtype if on_exists
3032 on_value = ops.convert_to_tensor(1, dtype, name="on_value")
3043 return gen_array_ops.one_hot(indices, depth, on_value, off_value, axis,
|
| D | ctc_ops.py | 440 on_value=0.0,
479 on_value=0.0, off_value=math_ops.log(0.0))
937 idx, depth=num_states, on_value=0.0, off_value=math_ops.log(0.0),
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/proto/ |
| D | tensor_forest_params.proto | 99 float on_value = 1; field
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | mixture_same_family.py | 254 on_value=np.ones([], dtype=npdt),
|
| D | mixture.py | 332 on_value=np.ones([], dtype=npdt),
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | beam_search_decoder_test.py | 401 on_value=True,
421 on_value=False,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers.py | 2539 on_value=1.0, argument
2561 labels, num_classes, on_value=on_value, off_value=off_value)
|
| D | feature_column.py | 1015 dense_id_tensor, depth=self.length, on_value=1.0, off_value=0.0)
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column.py | 2354 on_value=1.,
3043 on_value=1.0,
|
| D | feature_column_v2.py | 2893 on_value=1.,
4209 on_value=1.0,
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 2366 on_value = tf.placeholder(
2373 on_value=on_value,
2377 return [indices, depth, on_value, off_value], [out]
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 1338 on_value=count,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 793 …argspec: "args=[\'indices\', \'depth\', \'on_value\', \'off_value\', \'axis\', \'dtype\', \'name\'…
|