Home
last modified time | relevance | path

Searched refs:orig_broadcast_dims (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc1007 auto orig_broadcast_dims = consumer->dimensions(); in PropagateOnBroadcast() local
1010 absl::c_linear_search(orig_broadcast_dims, old_batch_dim); in PropagateOnBroadcast()