Searched defs:dynamic_dim (Results 1 – 4 of 4) sorted by relevance
90 def mksym(shape_env, value, source, dynamic_dim): argument
19 def mksym(shape_env, value, source, dynamic_dim) -> None: argument
1601 HloInstruction* dynamic_dim = in RewriteDynamicUpdateSlice() local1624 HloInstruction* dynamic_dim = dynamic_dims_in_partial_update[i]; in RewriteDynamicUpdateSlice() local
3296 def create_unspecified_symint_and_symbol(self, value, source, dynamic_dim): argument