Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast.h45 broadcasting_required_ = true; in MatMulBCast()
53 broadcasting_required_ = in MatMulBCast()
56 if (broadcasting_required_) { in MatMulBCast()
65 return !broadcasting_required_ || (batch_bcast_ && batch_bcast_->IsValid()); in IsValid()
67 bool IsBroadcastingRequired() const { return broadcasting_required_; } in IsBroadcastingRequired()
89 bool broadcasting_required_ = false; variable
Dbcast.h89 bool IsBroadcastingRequired() const { return broadcasting_required_; } in IsBroadcastingRequired()
113 bool broadcasting_required_ = true; variable
153 broadcasting_required_ = false; in BCastList()
304 if (return_flattened_batch_indices && broadcasting_required_ && in BCastList()