Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:script_ops

36 from tensorflow.python.ops import script_ops
56 z = self.evaluate(script_ops.py_func(np_func, [x, y], dtypes.float32))
64 script_ops.eager_py_func(np_func, [x, y], [dtypes.float32]))
71 z = self.evaluate(script_ops.py_func(np_func, [x, y], [dtypes.float64]))
79 z = self.evaluate(script_ops.py_func(np_func, [x, y], dtypes.complex64))
89 y = self.evaluate(script_ops.py_func(rfft, [x], dtypes.complex64))
99 y = self.evaluate(script_ops.py_func(literal, [x], dtypes.float64))
110 script_ops.py_func(list_func, [x], [dtypes.float64] * 2))
122 script_ops.py_func(tuple_func, [x], [dtypes.float64] * 2))
129 script_ops.py_func(tuple_func, (x,),
144 script_ops.py_func(read_fixed_length_numpy_strings, [],
147 script_ops.py_func(read_and_return_strings, [x, y], dtypes.string))
161 script_ops.py_func(read_fixed_length_numpy_strings, [],
164 script_ops.py_func(read_and_return_strings, [x, y], dtypes.string))
177 y, = script_ops.py_func(read_object_array, [],
179 z, = script_ops.py_func(read_and_return_strings, [x, y], [dtypes.string])
185 s, = script_ops.py_func(lambda: [correct], [], [dtypes.string])
192 s, = script_ops.py_func(lambda: [inp], [], [dtypes.string])
198 y = script_ops.py_func(lambda x: x + 1, [x], [dtypes.float32])
199 z = script_ops.py_func(lambda x: x * 2, [x], [dtypes.float32])
205 x = self.evaluate(script_ops.py_func(lambda: 42.0, [], dtypes.float64))
211 tf_array = script_ops.py_func(lambda: np_array, [], [dtypes.float32])
223 z, = script_ops.py_func(unicode_string, [], [dtypes.string])
233 y, = script_ops.py_func(bad, [], [dtypes.float32])
246 z, = script_ops.py_func(bad, [], [dtypes.int64])
261 z, = script_ops.py_func(ident, [p], [dtypes.float32])
271 x, = script_ops.py_func(lambda: next(producer), [], [dtypes.int64])
280 x, = script_ops.py_func(
290 x, = script_ops.py_func(lambda a: 0, [a], [dtypes.int64])
291 y, = script_ops.py_func(lambda a: 0, [a], [dtypes.int64], stateful=False)
298 x, = script_ops.py_func(lambda: np.array(val, order="F"), [],
317 x, = script_ops.py_func(blocking_put, [], [dtypes.int64])
318 y, = script_ops.py_func(blocking_get, [], [dtypes.int64])
334 return script_ops.py_func(self._increment, [diff], [], stateful=True)
352 f = script_ops.py_func(
365 f = script_ops.eager_py_func(raise_exception, [], [])
368 f = script_ops.eager_py_func(raise_exception, [], [])
370 f = script_ops.py_func(raise_exception, [], [])
394 _ = script_ops.py_func(lambda x: x + 1, [c], [dtypes.float32])
395 _ = script_ops.eager_py_func(lambda x: x + 1, [c], [dtypes.float32])
396 self.assertTrue(script_ops._py_funcs.size() < 100)
403 output = script_ops.eager_py_func(matmul, inp=[a, x], Tout=dtypes.int32)
413 output = script_ops.eager_py_func(matmul, inp=[a, x], Tout=dtypes.float32)
422 output = script_ops.eager_py_func(
433 output = script_ops.eager_py_func(no_return_value, inp=[], Tout=[])
446 return script_ops.eager_py_func(matmul, inp=[a, x], Tout=dtypes.float32)