Searched refs:swap_inputs (Results 1 – 5 of 5) sorted by relevance
42 ge.swap_inputs(c.op, [a, b])
95 reroute.swap_inputs(sgv, input_placeholders)
394 def swap_inputs(sgv0, sgv1): function
1763 for (auto swap_inputs : {false, true}) { in TestBinaryTensorOpWeightNoBroadcast()1766 if (swap_inputs) { in TestBinaryTensorOpWeightNoBroadcast()1779 /*values=*/swap_inputs ? operand1 : operand2); in TestBinaryTensorOpWeightNoBroadcast()1792 {"input", test::AsTensor<CType>(swap_inputs ? operand2 : operand1)}}; in TestBinaryTensorOpWeightNoBroadcast()
86 const bool swap_inputs = params->broadcast_category == in ProcessBroadcastShapes() local89 swap_inputs ? &extended_shape1 : &extended_shape0; in ProcessBroadcastShapes()91 swap_inputs ? &extended_shape0 : &extended_shape1; in ProcessBroadcastShapes()