Home
last modified time | relevance | path

Searched refs:swap_inputs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Delementwise.cc108 bool swap_inputs = false) { in GetTwoInputCode() argument
175 if (swap_inputs) { in GetTwoInputCode()
186 float scalar_parameter, bool swap_inputs) { in CreateElementwiseOneRuntimeOneScalar() argument
196 "second_val", swap_inputs); in CreateElementwiseOneRuntimeOneScalar()
206 bool swap_inputs) { in CreateElementwiseTwoInput() argument
227 "second_val", swap_inputs); in CreateElementwiseTwoInput()
237 bool swap_inputs) { in CreateElementwiseTwoInput() argument
261 "second_val", swap_inputs); in CreateElementwiseTwoInput()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dprocess_broadcast_shapes.h88 const bool swap_inputs = params->broadcast_category == in ProcessBroadcastShapes() local
91 swap_inputs ? &extended_shape1 : &extended_shape0; in ProcessBroadcastShapes()
93 swap_inputs ? &extended_shape0 : &extended_shape1; in ProcessBroadcastShapes()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc1547 for (bool swap_inputs : {false, true}) { in TEST_F()
1553 swap_inputs ? scale : inputs, in TEST_F()
1554 swap_inputs ? inputs : scale); in TEST_F()