Searched defs:index_of_constant_input (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dremove_trivial_binary.cc76 const int index_of_constant_input = is_input_constant[0] ? 0 : 1; local
78 CHECK(is_input_constant[index_of_constant_input]);
84 model->GetArray(binary_op->inputs[index_of_constant_input]);
95 is_trivial = index_of_constant_input == 1 &&
100 is_trivial = index_of_constant_input == 1 &&
H A Dresolve_multiply_by_zero.cc84 const int index_of_constant_input = is_input_constant[0] ? 0 : 1; local
86 CHECK(is_input_constant[index_of_constant_input]);
90 model->GetArray(mul_op->inputs[index_of_constant_input]);
H A Dfuse_binary_into_following_affine.cc33 int index_of_constant_input) {
36 CHECK(index_of_constant_input == 0 || index_of_constant_input == 1);
41 index_of_constant_input == 1);
49 model->GetArray(add_or_sub_op->inputs[index_of_constant_input]);
62 index_of_constant_input == 1) {
115 int index_of_constant_input) {
118 CHECK(index_of_constant_input == 0 || index_of_constant_input == 1);
122 index_of_constant_input
31 FuseAddOrSubParamsIntoFollowingAffine(Model* model, Operator* following_op, const Operator* add_or_sub_op, int index_of_constant_input) argument
113 FuseMulOrDivParamsIntoFollowingAffine(Model* model, Operator* following_op, const Operator* mul_or_div_op, int index_of_constant_input) argument
185 const int index_of_constant_input = is_input_constant[0] ? 0 : 1; local
[all...]
H A Dfuse_binary_into_preceding_affine.cc32 int index_of_constant_input) {
35 CHECK(index_of_constant_input == 0 || index_of_constant_input == 1);
42 model->GetArray(add_or_sub_op->inputs[index_of_constant_input]);
59 : (index_of_constant_input == 1)
80 int index_of_constant_input) {
83 CHECK(index_of_constant_input == 0 || index_of_constant_input == 1);
87 index_of_constant_input == 1);
98 model->GetArray(mul_or_div_op->inputs[index_of_constant_input]);
30 FuseAddOrSubParamsIntoPrecedingAffine(Model* model, Operator* preceding_op, const Operator* add_or_sub_op, int index_of_constant_input) argument
78 FuseMulOrDivParamsIntoPrecedingAffine(Model* model, Operator* preceding_op, const Operator* mul_or_div_op, int index_of_constant_input) argument
211 const int index_of_constant_input = is_input_constant[0] ? 0 : 1; local
[all...]

Completed in 127 milliseconds