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

Class ParamsProvider

dali/test/python/sequences_test_utils.py:147–172  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

145
146
147class ParamsProvider(ParamsProviderBase):
148 def __init__(self, input_params: List[ArgCb]):
149 super().__init__()
150 self.input_params = input_params
151 self.arg_input_data = None
152 self.expanded_params_data = None
153
154 def compute_params(self) -> List[ArgData]:
155 self.arg_input_data = compute_input_params_data(
156 self.input_data, self.rng, self.input_params
157 )
158 return self.arg_input_data
159
160 def expand_params(self) -> List[ArgData]:
161 self.expanded_params_data = [
162 ArgData(
163 desc=ArgDesc(arg_data.desc.name, "", arg_data.desc.dest_device),
164 data=expand_arg_input(self.input_data, arg_data),
165 )
166 for arg_data in self.arg_input_data
167 ]
168 return self.expanded_params_data
169
170 def __repr__(self):
171 class_name = repr(self.__class__.__name__).strip("'")
172 return f"{class_name}({repr(self.input_params)})"
173
174
175def arg_data_node(arg_data: ArgData):

Callers 1

_test_seq_inputFunction · 0.85

Calls

no outgoing calls

Tested by 1

_test_seq_inputFunction · 0.68