Searched refs:SpaceToBatchDimMap (Results 1 – 2 of 2) sorted by relevance
267 int DimMapper(SpaceToBatchDimMap s) { return static_cast<int>(s); } in DimMapper()363 old_dims[0] = dim_map_val[DimMapper(SpaceToBatchDimMap::kSpace0)]; in GetSpatialDimsToSplit()490 dim_map_val_op_0[DimMapper(SpaceToBatchDimMap::kBatch)]; in IsThisBackPropFilterConv()500 dim_map_val_op_0[DimMapper(SpaceToBatchDimMap::kBatch)]; in IsThisBackPropFilterConv()989 [DimMapper(SpaceToBatchDimMap::kBatch)] != in CanPropagate()991 [DimMapper(SpaceToBatchDimMap::kBatch)] || in CanPropagate()993 [DimMapper(SpaceToBatchDimMap::kSpace0)] != in CanPropagate()995 [DimMapper(SpaceToBatchDimMap::kSpace0)]) { in CanPropagate()1020 SpaceToBatchDimMap::kSpace0)])) { in CanPropagate()1084 dim_map[DimMapper(SpaceToBatchDimMap::kSpace0)]) { in CanPropagate()[all …]
41 enum class SpaceToBatchDimMap : uint8_t { enum