Searched defs:old_batch_dim (Results 1 – 1 of 1) sorted by relevance
826 const int64 old_batch_dim = dim_map_val_op_0.first; in CanPropagate() local861 const int64 old_batch_dim = dim_map_val_op_1.first; in CanPropagate() local963 const int64 old_batch_dim = dim_map_val_op_0.first; in CanPropagate() local1004 const int64 old_batch_dim = dim_map_val.first; in PropagateOnBroadcast() local1160 const int64 old_batch_dim = result.first; in SupportedOpForPropagation() local1257 const int64 old_batch_dim = result.first; in Propagate() local1366 const int64 old_batch_dim = dim_map_val.first; in Propagate() local1405 const int64 old_batch_dim = dim_map_val.first; in Propagate() local1652 int64 old_batch_dim, int64 old_space_dim) { in SelectValidPortion()1714 const int64 old_batch_dim = result.first; in BatchToSpace() local[all …]