Home
last modified time | relevance | path

Searched refs:convert_to_tensor (Results 1 – 25 of 333) sorted by relevance

12345678910>>...14

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/
H A Dnp_arrays_test.py39 a = ops.convert_to_tensor(value=1.1, dtype=dtypes.float32).astype(np.int32)
42 a = ops.convert_to_tensor(value=[0.0, 1.1], dtype=dtypes.float32).astype(
48 a = ops.convert_to_tensor(value=[1.0, 2.0])
52 a = ops.convert_to_tensor(value=a, dtype=np.int32)
53 b = ops.convert_to_tensor(value=b, dtype=np.int32)
55 out = ops.convert_to_tensor(value=out, dtype=np.int32)
108 ops.convert_to_tensor(value=[1.5, 1.25]),
114 ops.convert_to_tensor(value=[1.5, 1.25]),
119 a = ops.convert_to_tensor(value=a, dtype=np.int32)
120 b = ops.convert_to_tensor(value=b, dtype=np.int32)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Drandom_ops.py88 mean_tensor = ops.convert_to_tensor(mean, dtype=dtype, name="mean")
89 stddev_tensor = ops.convert_to_tensor(stddev, dtype=dtype, name="stddev")
139 means_tensor = ops.convert_to_tensor(means, dtype=dtype, name="means")
140 stddevs_tensor = ops.convert_to_tensor(stddevs, dtype=dtype, name="stddevs")
141 minvals_tensor = ops.convert_to_tensor(minvals, dtype=dtype, name="minvals")
142 maxvals_tensor = ops.convert_to_tensor(maxvals, dtype=dtype, name="maxvals")
197 mean_tensor = ops.convert_to_tensor(mean, dtype=dtype, name="mean")
198 stddev_tensor = ops.convert_to_tensor(stddev, dtype=dtype, name="stddev")
302 minval = ops.convert_to_tensor(minval, dtype=dtype, name="min")
303 maxval = ops.convert_to_tensor(maxval, dtype=dtype, name="max")
[all …]
H A Dmath_ops.py173 start = ops.convert_to_tensor(start, name="start")
175 stop = ops.convert_to_tensor(stop, name="stop", dtype=start.dtype)
405 x = ops.convert_to_tensor(x, name="x")
476 x = ops.convert_to_tensor(x, dtype=dtype)
619 scalar = ops.convert_to_tensor(
730 real = ops.convert_to_tensor(real, name="real")
731 imag = ops.convert_to_tensor(imag, name="imag")
779 x = ops.convert_to_tensor(x)
820 input = ops.convert_to_tensor(input, name="input")
855 input = ops.convert_to_tensor(input, name="input")
[all …]
H A Dstateless_random_ops.py198 seed = ops.convert_to_tensor(seed)
238 data = ops.convert_to_tensor(data)
311 seed = ops.convert_to_tensor(seed)
472 minval = ops.convert_to_tensor(minval, dtype=dtype, name="min")
473 maxval = ops.convert_to_tensor(maxval, dtype=dtype, name="max")
553 probs = ops.convert_to_tensor(
555 counts = ops.convert_to_tensor(
649 alpha = ops.convert_to_tensor(alpha, dtype=dtype, name="alpha")
650 beta = ops.convert_to_tensor(
763 mean = ops.convert_to_tensor(mean, dtype=dtype, name="mean")
[all …]
H A Dnn_impl.py85 log_input = ops.convert_to_tensor(log_input, name="log_input")
86 targets = ops.convert_to_tensor(targets, name="targets")
124 logits = ops.convert_to_tensor(logits, name="logits")
125 labels = ops.convert_to_tensor(labels, name="labels")
321 logits = ops.convert_to_tensor(logits, name="logits")
322 labels = ops.convert_to_tensor(labels, name="labels")
443 per_example_loss = ops.convert_to_tensor(per_example_loss)
448 sample_weight = ops.convert_to_tensor(sample_weight)
531 x = ops.convert_to_tensor(x, name="x")
532 weights = ops.convert_to_tensor(weights, name="weights")
[all …]
H A Dcheck_ops.py383 x = ops.convert_to_tensor(x, name='x')
384 y = ops.convert_to_tensor(y, name='y')
502 x = ops.convert_to_tensor(x, name='x')
512 zero = ops.convert_to_tensor(0, dtype=x.dtype)
556 x = ops.convert_to_tensor(x, name='x')
565 zero = ops.convert_to_tensor(0, dtype=x.dtype)
611 x = ops.convert_to_tensor(x, name='x')
621 zero = ops.convert_to_tensor(0, dtype=x.dtype)
667 x = ops.convert_to_tensor(x, name='x')
677 zero = ops.convert_to_tensor(0, dtype=x.dtype)
[all …]
H A Dstate_ops.py106 value = ops.convert_to_tensor(value, name="value")
109 init = ops.convert_to_tensor(init, name="init")
432 ref.handle, indices, ops.convert_to_tensor(updates, ref.dtype),
494 ref.handle, indices, ops.convert_to_tensor(updates, ref.dtype),
546 ref.handle, indices, ops.convert_to_tensor(updates, ref.dtype),
609 ref.handle, indices, ops.convert_to_tensor(updates, ref.dtype),
663 ref.handle, indices, ops.convert_to_tensor(updates, ref.dtype),
727 ref.handle, indices, ops.convert_to_tensor(updates, ref.dtype),
779 ref.handle, indices, ops.convert_to_tensor(updates, ref.dtype),
831 ref.handle, indices, ops.convert_to_tensor(updates, ref.dtype),
[all …]
H A Dimage_ops_impl.py504 image = ops.convert_to_tensor(image, name='image')
629 image = ops.convert_to_tensor(image, name='image')
686 image = ops.convert_to_tensor(image, name='image')
688 k = ops.convert_to_tensor(k, dtype=dtypes.int32, name='k')
823 image = ops.convert_to_tensor(image, name='image')
903 image = ops.convert_to_tensor(image, name='image')
1098 image = ops.convert_to_tensor(image, name='image')
1202 image = ops.convert_to_tensor(image, name='image')
1321 image = ops.convert_to_tensor(image, name='image')
1436 images = ops.convert_to_tensor(images, name='images')
[all …]
H A Darray_ops.py293 input = ops.convert_to_tensor(input)
686 input = ops.convert_to_tensor(input)
804 input = ops.convert_to_tensor(input)
807 return ops.convert_to_tensor(num_elements, dtype=out_type)
814 input = ops.convert_to_tensor(input)
877 input = ops.convert_to_tensor(input)
983 tensor = ops.convert_to_tensor(tensor)
1406 value_t = ops.convert_to_tensor(values[0])
1407 value_shape = ops.convert_to_tensor(value_t).get_shape()
1466 return ops.convert_to_tensor(values, name=name)
[all …]
H A Dimage_ops.py192 transforms = ops.convert_to_tensor(transforms, name="transforms")
225 transform_matrices = ops.convert_to_tensor(
246 image_or_images = ops.convert_to_tensor(images, name="images")
247 transform_or_transforms = ops.convert_to_tensor(
280 image_or_images = ops.convert_to_tensor(images, name="images")
281 transform_or_transforms = ops.convert_to_tensor(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/distributions/
H A Dutil.py58 x = ops.convert_to_tensor(x, name="x")
90 x = ops.convert_to_tensor(x, name="x")
114 a = ops.convert_to_tensor(a, name="a")
115 b = ops.convert_to_tensor(b, name="b")
194 logits = ops.convert_to_tensor(logits, name="logits", dtype=dtype)
205 probs = ops.convert_to_tensor(probs, name="probs", dtype=dtype)
340 x = ops.convert_to_tensor(categorical_param, name="categorical_param")
427 x = ops.convert_to_tensor(x, name="x")
510 n = ops.convert_to_tensor(n, name="n")
511 counts = ops.convert_to_tensor(counts, name="counts")
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/signal/
H A Dspectral_ops.py70 signals = ops.convert_to_tensor(signals, name='signals')
72 frame_length = ops.convert_to_tensor(frame_length, name='frame_length')
74 frame_step = ops.convert_to_tensor(frame_step, name='frame_step')
80 fft_length = ops.convert_to_tensor(fft_length, name='fft_length')
136 frame_step_ = ops.convert_to_tensor(frame_step, name='frame_step')
138 frame_length = ops.convert_to_tensor(frame_length, name='frame_length')
223 stfts = ops.convert_to_tensor(stfts, name='stfts')
225 frame_length = ops.convert_to_tensor(frame_length, name='frame_length')
227 frame_step = ops.convert_to_tensor(frame_step, name='frame_step')
232 fft_length = ops.convert_to_tensor(fft_length, name='fft_length')
[all …]
H A Dshape_ops.py116 signal = ops.convert_to_tensor(signal, name="signal")
117 frame_length = ops.convert_to_tensor(frame_length, name="frame_length")
118 frame_step = ops.convert_to_tensor(frame_step, name="frame_step")
119 axis = ops.convert_to_tensor(axis, name="axis")
155 pad_value = ops.convert_to_tensor(pad_value, signal.dtype)
169 ops.convert_to_tensor([[0, pad_samples]]),
218 outer_dimensions = ops.convert_to_tensor(outer_dimensions)
219 inner_dimensions = ops.convert_to_tensor(
221 mid_dimensions = ops.convert_to_tensor([num_frames, frame_length],
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/io_ops/
H A Dparsing_ops_test.py137 "serialized": ops.convert_to_tensor(["", ""]),
210 "serialized": ops.convert_to_tensor(serialized),
230 "serialized": ops.convert_to_tensor(serialized),
273 "serialized": ops.convert_to_tensor(serialized),
318 "serialized": ops.convert_to_tensor(serialized),
360 "serialized": ops.convert_to_tensor(serialized),
419 "serialized": ops.convert_to_tensor(serialized),
457 "serialized": ops.convert_to_tensor(serialized),
511 "serialized": ops.convert_to_tensor(serialized),
539 "serialized": ops.convert_to_tensor(serialized),
[all …]
H A Dparse_single_example_op_test.py144 "serialized": ops.convert_to_tensor(""),
206 "serialized": ops.convert_to_tensor(serialized),
223 "serialized": ops.convert_to_tensor(serialized),
270 "serialized": ops.convert_to_tensor(proto.SerializeToString()),
314 "serialized": ops.convert_to_tensor(proto.SerializeToString()),
349 "serialized": ops.convert_to_tensor(proto.SerializeToString()),
402 "serialized": ops.convert_to_tensor(proto.SerializeToString()),
445 "serialized": ops.convert_to_tensor(proto.SerializeToString()),
501 ops.convert_to_tensor(
527 "serialized": ops.convert_to_tensor(proto.SerializeToString()),
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dembedding_ops_test.py758 tf_embedding_sum = ops.convert_to_tensor(embedding_sum)
1054 ops.convert_to_tensor([0, 1, 2]),
1055 ops.convert_to_tensor([2, 3])
1058 ops.convert_to_tensor([12, 23, 34]),
1059 ops.convert_to_tensor([1, 2])
1068 ops.convert_to_tensor([0, 1, 2]),
1069 ops.convert_to_tensor([2, 3])
1072 ops.convert_to_tensor([12, 23, 34]),
1073 ops.convert_to_tensor([1, 2])
1082 ops.convert_to_tensor([0, 1, 2]),
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dlinear_operator_util_test.py33 x = ops.convert_to_tensor([0., 2, 3])
39 x = ops.convert_to_tensor([1., 0, 3])
40 y = ops.convert_to_tensor([0., 0, 0])
47 x = ops.convert_to_tensor([1., 2, 0])
48 y = ops.convert_to_tensor([1., 2, 0])
58 x = ops.convert_to_tensor([1., 2, 3])
65 x = ops.convert_to_tensor([1., 0, 3])
66 y = ops.convert_to_tensor([1., 2, 0])
74 x = ops.convert_to_tensor([1., 0, 3])
81 x = ops.convert_to_tensor([1., 2, 0])
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
H A Dparse_example_dataset_test.py133 ops.convert_to_tensor(["", ""]), {
196 ops.convert_to_tensor(serialized),
208 ops.convert_to_tensor(serialized),
247 ops.convert_to_tensor(serialized), {
286 ops.convert_to_tensor(serialized),
323 ops.convert_to_tensor(serialized), {
371 ops.convert_to_tensor(serialized), {
408 ops.convert_to_tensor(serialized), {
458 ops.convert_to_tensor(serialized), {
484 ops.convert_to_tensor(serialized), {
[all …]
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dserve_slices.py31 tf.convert_to_tensor(subvalue, dtype=dtype)
82 callback_token=tf.convert_to_tensor(callback_token, dtype=tf.string),
84 max_key=tf.convert_to_tensor(max_key, dtype=tf.int32),
85 select_fn_initialize_op=tf.convert_to_tensor(
89 select_fn_key_input_tensor_name=tf.convert_to_tensor(
91 select_fn_filename_input_tensor_name=tf.convert_to_tensor(
93 select_fn_target_tensor_name=tf.convert_to_tensor(
H A Dexample_selector_fuser_test.py31 tf.convert_to_tensor(selector.SerializeToString(), dtype=tf.string),
32 tf.convert_to_tensor(
34 tf.convert_to_tensor(
52 tf.convert_to_tensor(b'1234', dtype=tf.string),
53 tf.convert_to_tensor(
55 tf.convert_to_tensor(
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
H A Dstack_op_test.py306 ops.convert_to_tensor([1, constant_op.constant(0), 2]))
308 ops.convert_to_tensor([[0, 0, 0],
313 ops.convert_to_tensor([[0, 0, 0],
317 ops.convert_to_tensor([
325 result = ops.convert_to_tensor([[[0., 0.],
334 t_0 = ops.convert_to_tensor([[0., 0., 0.], [0., 0., 0.], [0., 0., 0.]])
337 t_1 = ops.convert_to_tensor([[0., 0., 0.], constant_op.constant(
341 t_2 = ops.convert_to_tensor(
345 t_3 = ops.convert_to_tensor(
352 t_4 = ops.convert_to_tensor(
[all …]
H A Dconstant_op_test.py44 tf_ans = ops.convert_to_tensor(x).eval()
54 tf_ans = ops.convert_to_tensor(x).eval()
172 val = ops.convert_to_tensor(b"\0\0\0\0").eval()
177 val = ops.convert_to_tensor(b"xx\0xx").eval()
183 val = ops.convert_to_tensor(nested).eval()
291 x = ops.convert_to_tensor(t)
296 x = ops.convert_to_tensor(10.0)
301 x = ops.convert_to_tensor(tensor_shape.TensorShape([]))
305 x = ops.convert_to_tensor(tensor_shape.TensorShape([1, 2, 3]))
309 x = ops.convert_to_tensor(tensor_shape.TensorShape([2**31-1, 2, 3]))
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
H A Dcwise_ops_test.py85 ops.convert_to_tensor(np.array([x]).astype(dtype)),
86 ops.convert_to_tensor(np.array([y]).astype(dtype)))
121 out = tf_func(ops.convert_to_tensor(x), ops.convert_to_tensor(y))
226 inx = ops.convert_to_tensor(x)
227 iny = ops.convert_to_tensor(y)
237 out = math_ops.logical_not(ops.convert_to_tensor(x))
345 inx = ops.convert_to_tensor(x)
346 iny = ops.convert_to_tensor(y)
358 inxf = ops.convert_to_tensor(xf)
359 inyf = ops.convert_to_tensor(yf)
[all …]
/aosp_15_r20/external/libopus/dnn/torch/weight-exchange/wexchange/tf/
H A Dtf.py85 gru.weights[0].assign(tf.convert_to_tensor(w_ih.transpose()))
86 gru.weights[1].assign(tf.convert_to_tensor(w_hh.transpose()))
87 gru.weights[2].assign(tf.convert_to_tensor(np.vstack((b_ih, b_hh))))
115 dense.weights[0].assign(tf.convert_to_tensor(w))
117 dense.weights[1].assign(tf.convert_to_tensor(b))
147 conv.weights[0].assign(tf.convert_to_tensor(w))
149 conv.weights[1].assign(tf.convert_to_tensor(b))
163 emb.weights[0].assign(tf.convert_to_tensor(w))
/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/simple_hash_table/
H A Dsimple_hash_table.py71 self._default_value = tf.convert_to_tensor(default_value, dtype=value_dtype)
139 key = tf.convert_to_tensor(key, dtype=self._key_dtype, name="key")
141 dynamic_default_value = tf.convert_to_tensor(
166 key = tf.convert_to_tensor(key, self._key_dtype, name="key")
167 value = tf.convert_to_tensor(value, self._value_dtype, name="value")
187 key = tf.convert_to_tensor(key, self._key_dtype, name="key")

12345678910>>...14