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

Method _init_arithm_op

dali/python/nvidia/dali/_debug_mode.py:456–467  ·  view source on GitHub ↗

Fills arithmetic operator init arguments and returns inputs that are DataNodes.

(self, name, inputs)

Source from the content-addressed store, hash-verified

454 self._kwargs_classification[key] = classification
455
456 def _init_arithm_op(self, name, inputs):
457 """Fills arithmetic operator init arguments and returns inputs that are DataNodes."""
458
459 categories_idxs, data_nodes, integers, reals = _ops._group_inputs(inputs)
460 input_desc = _ops._generate_input_desc(categories_idxs, integers, reals)
461
462 self._init_args["device"] = _ops._choose_device(data_nodes)
463 self._init_args["expression_desc"] = f"{name}({input_desc})"
464 self._init_args["integer_constants"] = integers
465 self._init_args["real_constants"] = reals
466
467 return data_nodes
468
469 def _pack_to_data_node_debug(self, data, position=None):
470 if isinstance(data, (list, tuple)):

Callers 1

__init__Method · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected