Searched refs:kernel_mapping_scheme (Results 1 – 2 of 2) sorted by relevance
4116 const KernelMappingScheme& kernel_mapping_scheme) { in GetUnnormalizedIndex() argument4132 kernel_mapping_scheme.GetDimsInElems(), b_); in GetUnnormalizedIndex()4364 const KernelMappingScheme& kernel_mapping_scheme = in GetUntransposedOutputLinearAddress() local4371 absl::Span<const int64> dims_in_elem = kernel_mapping_scheme.GetDimsInElems(); in GetUntransposedOutputLinearAddress()
246 "kernel_mapping_scheme.h",