Searched refs:index_of_constant_input (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
H A D | remove_trivial_binary.cc | 76 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 D | fuse_binary_into_following_affine.cc | 33 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 D | fuse_binary_into_preceding_affine.cc | 32 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...] |
H A D | resolve_multiply_by_zero.cc | 84 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]);
|
Completed in 242 milliseconds