Searched defs:space_dim (Results 1 – 1 of 1) sorted by relevance
1518 const int64_t space_dim = result[DimMapper(SpaceToBatchDimMap::kSpace0)]; in IsBroadcastPropagatable() local1633 const int64_t space_dim = result[DimMapper(SpaceToBatchDimMap::kSpace0)]; in SupportedOpForPropagation() local1804 const int64_t space_dim = DimLookUp(permute_dims, old_space_dim); in Propagate() local2390 const int64_t space_dim = DimLookUp(permute_dims, old_space_dim); in BatchToSpace() local