MCPcopy Create free account
hub / github.com/NVIDIA/DALI / test_python_operator_error

Function test_python_operator_error

dali/test/python/test_dali_tf_plugin_run.py:176–180  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

174 glob="Note that some operators * cannot be used with TensorFlow Dataset API and DALIIterator",
175)
176def test_python_operator_error():
177 daliop = dali_tf.DALIIterator()
178 pipe = PythonOperatorPipeline()
179 with tf.device("/cpu:0"):
180 _ = daliop(pipeline=pipe, shapes=[(1, 3, 3, 3)], dtypes=[tf.float32], device_id=0)

Callers

nothing calls this directly

Calls 2

deviceMethod · 0.45

Tested by

no test coverage detected