Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast.h41 const int64 y_batch_size() const { return y_batch_size_; } in y_batch_size()
61 int64 y_batch_size_; variable
Dmatmul_bcast.cc28 y_batch_size_ = TensorShape(batch_bcast_->y_reshape()).num_elements(); in MatMulBCast()
32 std::min(x_batch_size_, y_batch_size_) != output_batch_size_; in MatMulBCast()