Searched defs:permute_dims (Results 1 – 1 of 1) sorted by relevance
199 int64 DimLookUp(absl::Span<const int64> permute_dims, int64 id) { in DimLookUp()203 int64 ReverseDimLookUp(absl::Span<const int64> permute_dims, int64 id) { in ReverseDimLookUp()728 auto permute_dims = instr_to_dim_permute_map_[new_instr]; in CanPropagate() local1001 auto permute_dims = instr_to_dim_permute_map_[new_producer]; in PropagateOnBroadcast() local1183 auto permute_dims = instr_to_dim_permute_map_[new_operand]; in SupportedOpForPropagation() local1265 auto permute_dims = instr_to_dim_permute_map_[new_instr]; in Propagate() local1368 auto permute_dims = instr_to_dim_permute_map_[first_operand]; in Propagate() local1407 auto permute_dims = instr_to_dim_permute_map_[first_operand]; in Propagate() local1726 auto permute_dims = instr_to_dim_permute_map_[new_instr]; in BatchToSpace() local1850 auto permute_dims = instr_to_dim_permute_map_[activations_new]; in PropagateOnConv() local[all …]