Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast.h52 output_batch_size_ = output_batch_shape_.num_elements(); in MatMulBCast()
54 std::min(x_batch_size_, y_batch_size_) != output_batch_size_; in MatMulBCast()
57 ComputeBatchIndices(output_batch_size_, batch_bcast_->x_reshape(), in MatMulBCast()
59 ComputeBatchIndices(output_batch_size_, batch_bcast_->y_reshape(), in MatMulBCast()
69 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size()
93 int64 output_batch_size_ = 1; variable
Dbcast.h99 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size()
120 int64 output_batch_size_; variable
143 output_batch_size_ = 1; in BCastList()
166 output_batch_size_ = elements; in BCastList()
227 output_batch_size_ = mul_dims(output_batch_size_, output_.back()); in BCastList()
305 output_batch_size_ > 0) { in BCastList()
307 ComputeBatchIndices(output_batch_size_, reshape_[i], bcast_[i], in BCastList()