Searched defs:is_batch (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/cc/gradients/
H A Dmath_grad_test.cc577 void TestMatMulGrad(const bool is_batch, const bool t_x, const bool t_y) { argument
581 RandMatMulShapes(is_batch, t_x, t_y, &shapes);
590 if (is_batch) {
602 void RandMatMulShapes(const bool is_batch, const bool tx, const bool ty, argument
612 if (is_batch) {
622 if (is_batch) {
632 if (is_batch) {
H A Dmath_grad.cc1011 Status MatMulGradHelper(const Scope& scope, const bool is_batch, argument
1016 if (is_batch == false) {
1038 const bool is_batch,
1045 if (is_batch == false) {
1057 return MatMulGradHelper(scope, is_batch, grad_inputs[0], false, b, true, a,
1060 return MatMulGradHelper(scope, is_batch, grad_inputs[0], false, b, false,
1063 return MatMulGradHelper(scope, is_batch, b, false, grad_inputs[0], true, a,
1066 return MatMulGradHelper(scope, is_batch, b, true, grad_inputs[0], true,
1037 MatMulGradCommon(const Scope& scope, const Operation& op, const bool is_batch, const std::vector<Output>& grad_inputs, const string& attr_adj_x, const string& attr_adj_y, std::vector<Output>* grad_outputs) argument

Completed in 428 milliseconds