Searched refs:inner_kwargs (Results 1 – 3 of 3) sorted by relevance
71 inner_kwargs = _replace_ragged_with_flat_values(kwargs, nested_splits_lists)80 op(*inner_args, **inner_kwargs), nested_splits_lists[0])
426 def _call_next_creator_renaming_initializer(initializer, **inner_kwargs): argument427 inner_kwargs.pop("name") # Ignored; this is the scope-stripped name which432 **inner_kwargs)
1074 def enumerated_fn(*inner_args, **inner_kwargs): argument1075 r = map_fn(ix[0], *inner_args, **inner_kwargs)