Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast.h40 const int64 x_batch_size() const { return x_batch_size_; } in x_batch_size()
60 int64 x_batch_size_; variable
Dmatmul_bcast.cc27 x_batch_size_ = TensorShape(batch_bcast_->x_reshape()).num_elements(); in MatMulBCast()
32 std::min(x_batch_size_, y_batch_size_) != output_batch_size_; in MatMulBCast()