Searched refs:a_times_b (Results 1 – 3 of 3) sorted by relevance
329 a_times_b = math_ops.matmul(339 return reshape_inv(a_times_b)
280 def a_times_b(inputs): function285 out = a_times_b(pair({'a': t}, {'b': t}))455 def a_times_b(inputs): function459 sq_op = a_times_b.get_concrete_function(
2330 auto a_times_b = computation_->AddInstruction(HloInstruction::CreateBinary( in HandlePower() local2334 a_times_b)); in HandlePower()