Home
last modified time | relevance | path

Searched refs:CountOpsWithInput (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_range.cc109 CountOpsWithInput(*model, op->inputs[0]) == 1) { in Run()
113 CountOpsWithInput(*model, op->inputs[1]) == 1) { in Run()
117 CountOpsWithInput(*model, op->inputs[2]) == 1) { in Run()
Dresolve_tensorflow_matmul.cc171 if (CountOpsWithInput(*model, previous_op_output) == 1) { in Run()
177 if (CountOpsWithInput(*model, previous_op_output) == 1) { in Run()
179 if (CountOpsWithInput(*model, previous_op_shape) == 1 && in Run()
Dremove_unused_op.cc71 CountOpsWithInput(*model, rnn_state.state_array())) { in Run()
76 if (CountOpsWithInput(*model, output)) { in Run()
Dreorder_reshape_transpose.cc146 if (CountOpsWithInput(*model, intermediate_name) != 1) { in Run()
210 if (CountOpsWithInput(*model, transpose_op->inputs[1]) != 1) { in Run()
228 if (CountOpsWithInput(*model, reshape_op->inputs[1]) != 1) { in Run()
Dfuse_binary_into_preceding_affine.cc296 const int count_ops_consuming_bias = CountOpsWithInput(*model, bias_name); in Run()
298 CountOpsWithInput(*model, weights_name); in Run()
334 CountOpsWithInput(*model, preceding_op->outputs[0]); in Run()
366 if (CountOpsWithInput(*model, old_constant_param_name) == 1) { in Run()
Dresolve_constant_fill.cc114 CountOpsWithInput(*model, op->inputs[0]) == 1) { in Run()
118 CountOpsWithInput(*model, op->inputs[1]) == 1) { in Run()
Dresolve_squeeze_attributes.cc40 if (CountOpsWithInput(*model, squeeze_op->outputs[0]) == 1) { in Run()
Dremove_tensorflow_assert.cc49 CHECK(!CountOpsWithInput(*model, assert_op->outputs[0])); in Run()
Dremove_final_dequantize_op.cc43 if (CountOpsWithInput(*model, output)) { in Run()
Ddrop_fake_quant.cc49 if (CountOpsWithInput(*model, fakequant_op->inputs[i]) == 1) { in Run()
Dremove_trivial_concatenation_input.cc64 CountOpsWithInput(*model, input) == 1) { in Run()
Dresolve_constant_shape_or_rank.cc66 CountOpsWithInput(*model, op->inputs[0]) == 1) { in Run()
Dremove_trivial_reshape.cc59 if (CountOpsWithInput(model, op.outputs[0]) == 1) { in IsReshapeTrivial()
Dfuse_binary_into_following_affine.cc221 if (CountOpsWithInput(*model, binary_op->outputs[0]) != 1) { in Run()
302 if (CountOpsWithInput(*model, old_constant_param_name) == 1) { in Run()
Dmake_initial_dequantize_operator.cc49 if (CountOpsWithInput(*model, input_name) > 1) { in AddDequantizeOperatorToInput()
Dconvert_trivial_tile_to_concat.cc84 CountOpsWithInput(*model, tile_op->inputs[1]) == 1) { in Run()
Dresolve_tensorflow_concat.cc80 if (CountOpsWithInput(*model, axis_name) == 1) { in Run()
Dconvert_expanddims_to_reshape.cc85 CountOpsWithInput(*model, axis_array_name) == 1 && in Run()
Dresolve_tensorflow_switch.cc117 if (CountOpsWithInput(*model, input) == 1 && in Run()
Dfuse_activation_functions.cc56 int count_ops_consuming_output = CountOpsWithInput(*model, ac_op->inputs[0]); in Run()
Dresolve_constant_binary.cc250 if (CountOpsWithInput(*model, binary_op->inputs[0]) == 1) { in Run()
253 if (CountOpsWithInput(*model, binary_op->inputs[1]) == 1) { in Run()
Dconvert_trivial_transpose_to_reshape.cc105 CountOpsWithInput(*model, perm_array_name) == 1) { in Run()
Dresolve_constant_reshape.cc114 CountOpsWithInput(*model, input) == 1) { in Run()
Dconvert_pure_conv_to_depthwise.cc56 if (CountOpsWithInput(*model, weights_name) > 1) { in Run()
Dreorder_elementwise_unary.cc90 if (CountOpsWithInput(*model, intermediate_name) != 1) { in Run()

12