Searched defs:is_batch (Results 1 – 4 of 4) sorted by relevance
1047 Status MatMulGradHelper(const Scope& scope, const bool is_batch, in MatMulGradHelper()1074 const bool is_batch, in MatMulGradCommon()
575 void TestMatMulGrad(const bool is_batch, const bool t_x, const bool t_y) { in TestMatMulGrad()600 void RandMatMulShapes(const bool is_batch, const bool tx, const bool ty, in RandMatMulShapes()
694 bool is_batch) { in KeepRunningInternal()