Searched refs:DimMapper (Results 1 – 1 of 1) sorted by relevance
267 int DimMapper(SpaceToBatchDimMap s) { return static_cast<int>(s); } in DimMapper() function in xla::__anon5e1d62920111::ConvolutionVisitor363 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()1019 j == instr_to_dim_map_[pivot_operand][DimMapper( in CanPropagate()1084 dim_map[DimMapper(SpaceToBatchDimMap::kSpace0)]) { in CanPropagate()[all …]