Searched refs:output_batch_shape_ (Results 1 – 1 of 1) sorted by relevance
51 output_batch_shape_ = TensorShape(batch_bcast_->output_shape()); in MatMulBCast()52 output_batch_size_ = output_batch_shape_.num_elements(); in MatMulBCast()72 const TensorShape& output_batch_shape() const { return output_batch_shape_; } in output_batch_shape()92 TensorShape output_batch_shape_; variable