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

Function explicit_conditional_ops_pipeline

dali/test/python/test_dali_cpu_only.py:1387–1397  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

1385
1386 @pipeline_def
1387 def explicit_conditional_ops_pipeline():
1388 value = types.Constant(np.array([42]), device="cpu")
1389 pred = fn.random.coin_flip(dtype=types.DALIDataType.BOOL)
1390 pred_validated = fn._conditional.validate_logical(
1391 pred, expression_name="or", expression_side="right"
1392 )
1393 true, false = fn._conditional.split(value, predicate=pred)
1394 true = true + 10
1395 merged = fn._conditional.merge(true, false, predicate=pred)
1396 negated = fn._conditional.not_(pred)
1397 return merged, negated, pred_validated
1398
1399 pipe = explicit_conditional_ops_pipeline(batch_size=5, num_threads=1, device_id=None)
1400 pipe.run()

Callers 1

test_conditionalFunction · 0.85

Calls 4

ConstantMethod · 0.80
splitMethod · 0.80
mergeMethod · 0.45
not_Method · 0.45

Tested by

no test coverage detected