| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | remove_unused_op.cc | 71 CountOpsWithInput(*model, rnn_state.state_array())) { in Run()
76 if (CountOpsWithInput(*model, output)) { in Run()
|
| D | remove_successive_transpose.cc | 69 if (CountOpsWithInput(*model, t_op->outputs[0]) != 1) { in Run()
78 if (!CountOpsWithInput(*model, t_next->outputs[0])) { in Run()
|
| D | reorder_reshape_transpose.cc | 146 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()
|
| D | resolve_tensorflow_matmul.cc | 208 if (CountOpsWithInput(*model, previous_op_output) == 1) { in Run()
215 if (CountOpsWithInput(*model, previous_op_output) == 1) { in Run()
|
| D | resolve_squeeze_attributes.cc | 40 if (CountOpsWithInput(*model, squeeze_op->outputs[0]) == 1) { in Run()
|
| D | remove_final_dequantize_op.cc | 43 if (CountOpsWithInput(*model, output)) { in Run()
|
| D | remove_tensorflow_assert.cc | 49 CHECK(!CountOpsWithInput(*model, assert_op->outputs[0])); in Run()
|
| D | drop_fake_quant.cc | 49 if (CountOpsWithInput(*model, fakequant_op->inputs[i]) == 1) { in Run()
|
| D | fuse_binary_into_preceding_affine.cc | 320 const int count_ops_consuming_bias = CountOpsWithInput(*model, bias_name); in Run()
322 CountOpsWithInput(*model, weights_name); in Run()
358 CountOpsWithInput(*model, preceding_op->outputs[0]); in Run()
|
| D | remove_trivial_reshape.cc | 59 if (CountOpsWithInput(model, op.outputs[0]) == 1) { in IsReshapeTrivial()
|
| D | convert_trivial_tile_to_concat.cc | 84 CountOpsWithInput(*model, tile_op->inputs[1]) == 1) { in Run()
|
| D | make_initial_dequantize_operator.cc | 50 if (CountOpsWithInput(*model, input_name) > 1) { in AddDequantizeOperatorToInput()
|
| D | resolve_tensorflow_switch.cc | 126 if (CountOpsWithInput(*model, input) == 1 && in Run()
|
| D | fuse_activation_functions.cc | 56 int count_ops_consuming_output = CountOpsWithInput(*model, ac_op->inputs[0]); in Run()
|
| D | convert_expanddims_to_reshape.cc | 92 CountOpsWithInput(*model, axis_array_name) == 1 && in Run()
|
| D | convert_trivial_transpose_to_reshape.cc | 106 CountOpsWithInput(*model, perm_array_name) == 1) { in Run()
|
| D | convert_pure_conv_to_depthwise.cc | 56 if (CountOpsWithInput(*model, weights_name) > 1) { in Run()
|
| D | reorder_elementwise_unary.cc | 91 if (CountOpsWithInput(*model, intermediate_name) != 1) { in Run()
|
| D | shuffle_fc_weights.cc | 109 if (CountOpsWithInput(*model, weights_name) != 1) { in Run()
|
| D | merge_reshape_into_preceding_transpose.cc | 124 if (CountOpsWithInput(*model, intermediate_name) != 1) { in Run()
|
| D | move_binary_operator_before_reshape.cc | 153 CountOpsWithInput(*model, binary_op->inputs[variable_input_idx]); in Run()
|
| D | fuse_binary_into_following_affine.cc | 221 if (CountOpsWithInput(*model, binary_op->outputs[0]) != 1) { in Run()
|
| D | quantize.cc | 593 if (!CountOpsWithInput(*model, dequantize_op->outputs[0])) { in Run()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 66 int CountOpsWithInput(const Model& model, const std::string& array_name);
|
| D | tooling_util.cc | 142 int CountOpsWithInput(const Model& model, const std::string& array_name) { in CountOpsWithInput() function
160 CountOpsWithInput(*model, array_name) == 0 && in DeleteArrayIfUnused()
173 if (CountOpsWithInput(*model, array_name) > 1) { in DeleteArrayIfUnusedOutsideOfOp()
2426 QCHECK_EQ(CountOpsWithInput(*model, weights_name), 1); in UndoWeightsShuffling()
|