| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| create_im2col_arrays.cc | 53 AvailableArrayName(*model, op->inputs[0] + "_im2col");
68 const string& im2col_array_name = AvailableArrayName(
|
| unroll_batch_matmul.cc | 57 slice_a_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_a")};
69 AvailableArrayName(*model, batch_name + "/slice_a/reshape")};
89 slice_b_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_b")};
101 AvailableArrayName(*model, batch_name + "/slice_b/reshape")};
111 matmul_op->outputs = {AvailableArrayName(*model, batch_name)};
151 pack_op->outputs = {AvailableArrayName(*model, batch_name + "/pack")};
189 model, AvailableArrayName(*model, input + "/transpose/perm"), perm_array);
192 transpose_op->outputs = {AvailableArrayName(*model, input + "/transpose")};
|
| resolve_batch_normalization.cc | 66 AvailableArrayName(*model, bn_op->outputs[0] + "_mul");
68 AvailableArrayName(*model, bn_op->outputs[0] + "_add");
69 const string mul_param_name = AvailableArrayName(*model, mul_name + "_param");
70 const string add_param_name = AvailableArrayName(*model, add_name + "_param");
|
| unfuse_activation_functions.cc | 68 AvailableArrayName(*model, op->outputs[0] + "_unfused");
|
| group_bidirectional_sequence_ops.cc | 201 AvailableArrayName(*model, "bidirectional_sequence_lstm_input_0");
235 const string& temp_array_name = AvailableArrayName(
244 AvailableArrayName(*model, "bidirectional_sequence_lstm_fw_output_0");
246 AvailableArrayName(*model, "bidirectional_sequence_lstm_bw_output_0");
264 AvailableArrayName(*model, "bidirectional_sequence_rnn_input_0");
283 const string& temp_array_name = AvailableArrayName(
292 AvailableArrayName(*model, "bidirectional_sequence_rnn_fw_output_0");
294 AvailableArrayName(*model, "bidirectional_sequence_rnn_bw_output_0");
373 const string& final_unpack_output_array_name = AvailableArrayName(
|
| convert_reorder_axes.cc | 45 AvailableArrayName(*model, reshape_op->outputs[0]);
74 string perm_array_name = AvailableArrayName(*model, transpose_op->outputs[0]);
|
| resolve_tensorflow_matmul.cc | 103 model, AvailableArrayName(*model, input_rhs + "/transpose/perm"),
106 AvailableArrayName(*model, input_rhs + "/transpose")};
|
| convert_trivial_pack_to_reshape.cc | 63 AvailableArrayName(*model, pack_op->outputs[0] + "_shape");
|
| ensure_bias_vectors.cc | 53 const string& bias_name = AvailableArrayName(*model, output_name + "_bias");
|
| make_initial_dequantize_operator.cc | 66 AvailableArrayName(*model, input_name + "_dequantized");
|
| identify_lstm_merge_inputs.cc | 77 string merged_weights = AvailableArrayName(*model, base_name + "weights");
120 string merged_biases = AvailableArrayName(*model, base_name + "biases");
164 AvailableArrayName(*model, base_name + "concat_temp");
|
| reorder_reshape_transpose.cc | 186 AvailableArrayName(*model, transpose_op->outputs[0] + "_exchange");
212 AvailableArrayName(*model, transpose_op->inputs[1] + "_copy");
230 AvailableArrayName(*model, reshape_op->inputs[1] + "_copy");
|
| unpartition_embedding_lookup.cc | 180 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_unpartitioned"));
190 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_permuted/perm"));
192 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_permuted"));
|
| convert_expanddims_to_reshape.cc | 74 string shape_array_name = toco::AvailableArrayName(*model, axis_array_name);
|
| convert_trivial_transpose_to_reshape.cc | 94 string shape_array_name = toco::AvailableArrayName(*model, perm_array_name);
|
| identify_prelu.cc | 111 AvailableArrayName(*model, neg_alpha_tensor_name + "_neg");
|
| lstm_utils.cc | 44 *array_name = AvailableArrayName(*model, *array_name);
|
| reorder_elementwise_unary.cc | 118 AvailableArrayName(*model, element_op->outputs[0] + "_reorder");
|
| resolve_reorder_axes.cc | 32 const string& newname = AvailableArrayName(*model, desired_newname);
|
| shuffle_fc_weights.cc | 147 AvailableArrayName(*model, fc_op->inputs[0] + "_shuffled");
|
| identify_lstm.cc | 278 AvailableArrayName(*model, base_name + "concat_temp");
284 AvailableArrayName(*model, base_name + "activ_temp");
|
| dequantize.cc | 155 const string& new_array_name = AvailableArrayName(*model, array_name);
|
| quantize.cc | 645 AvailableArrayName(*model, output + "_dequantized");
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| import.cc | 154 toco::AvailableArrayName(*model, "OptionalTensor");
|
| /external/tensorflow/tensorflow/lite/toco/ |
| tooling_util.h | 256 string AvailableArrayName(const Model& model, const string& name);
|
