Home
last modified time | relevance | path

Searched defs:old_batch_dim (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc489 const int64_t old_batch_dim = in IsThisBackPropFilterConv() local
499 const int64_t old_batch_dim = in IsThisBackPropFilterConv() local
1202 const int64_t old_batch_dim = in CanPropagate() local
1245 const int64_t old_batch_dim = in CanPropagate() local
1368 const int64_t old_batch_dim = in CanPropagate() local
1412 const int64_t old_batch_dim = in PropagateOnBroadcast() local
1569 const int64_t old_batch_dim = result[DimMapper(SpaceToBatchDimMap::kBatch)]; in SupportedOpForPropagation() local
1586 const int64_t old_batch_dim = result[DimMapper(SpaceToBatchDimMap::kBatch)]; in SupportedOpForPropagation() local
1609 const int64_t old_batch_dim = result[DimMapper(SpaceToBatchDimMap::kBatch)]; in SupportedOpForPropagation() local
1657 const int64_t old_batch_dim = result[DimMapper(SpaceToBatchDimMap::kBatch)]; in SupportedOpForPropagation() local
[all …]