Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast.cc30 output_batch_size_ = output_shape_.num_elements(); in MatMulBCast()
32 std::min(x_batch_size_, y_batch_size_) != output_batch_size_; in MatMulBCast()
35 ComputeBatchIndices(output_batch_size_, batch_bcast_->x_reshape(), in MatMulBCast()
37 ComputeBatchIndices(output_batch_size_, batch_bcast_->y_reshape(), in MatMulBCast()
Dbcast.h99 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size()
120 int64 output_batch_size_; variable
137 output_batch_size_ = 1; in BCastList()
160 output_batch_size_ = elements; in BCastList()
221 output_batch_size_ *= output_.back(); in BCastList()
298 output_batch_size_ > 0) { in BCastList()
300 ComputeBatchIndices(output_batch_size_, reshape_[i], bcast_[i], in BCastList()
Dmatmul_bcast.h39 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size()
63 int64 output_batch_size_; variable