| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 51 reshape_op->outputs = {AvailableArrayName( in SliceInput()
70 AvailableArrayName(*model, absl::StrCat(batch_name, "/slice"))}; in SliceInput()
82 AvailableArrayName(*model, absl::StrCat(batch_name, "/reshape"))}; in SliceInput()
118 model, AvailableArrayName(*model, input + "/transpose/perm"), perm_array); in TransposeInput()
121 transpose_op->outputs = {AvailableArrayName(*model, input + "/transpose")}; in TransposeInput()
224 matmul_op->outputs = {AvailableArrayName(*model, batch_name)}; in Run()
236 pack_op->outputs = {AvailableArrayName(*model, base_name + "/pack")}; in Run()
|
| D | resolve_tensorflow_matmul.cc | 112 model, AvailableArrayName(*model, input_lhs + "/transpose/perm"), in Run()
115 AvailableArrayName(*model, input_lhs + "/transpose")}; in Run()
147 model, AvailableArrayName(*model, input_rhs + "/transpose/perm"), in Run()
150 AvailableArrayName(*model, input_rhs + "/transpose")}; in Run()
|
| D | resolve_batch_normalization.cc | 66 AvailableArrayName(*model, bn_op->outputs[0] + "_mul"); in Run()
68 AvailableArrayName(*model, bn_op->outputs[0] + "_add"); in Run()
70 AvailableArrayName(*model, mul_name + "_param"); in Run()
72 AvailableArrayName(*model, add_name + "_param"); in Run()
|
| D | group_bidirectional_sequence_ops.cc | 201 AvailableArrayName(*model, "bidirectional_sequence_lstm_input_0"); in ConstructBidirectionalSequenceOp()
235 const std::string& temp_array_name = AvailableArrayName( in ConstructBidirectionalSequenceOp()
244 AvailableArrayName(*model, "bidirectional_sequence_lstm_fw_output_0"); in ConstructBidirectionalSequenceOp()
246 AvailableArrayName(*model, "bidirectional_sequence_lstm_bw_output_0"); in ConstructBidirectionalSequenceOp()
264 AvailableArrayName(*model, "bidirectional_sequence_rnn_input_0"); in ConstructBidirectionalSequenceOp()
283 const std::string& temp_array_name = AvailableArrayName( in ConstructBidirectionalSequenceOp()
292 AvailableArrayName(*model, "bidirectional_sequence_rnn_fw_output_0"); in ConstructBidirectionalSequenceOp()
294 AvailableArrayName(*model, "bidirectional_sequence_rnn_bw_output_0"); in ConstructBidirectionalSequenceOp()
375 const std::string& final_unpack_output_array_name = AvailableArrayName( in RewireFinalUnpackOutputs()
|
| D | create_im2col_arrays.cc | 53 AvailableArrayName(*model, op->inputs[0] + "_im2col"); in ProcessConvOperator()
68 const std::string& im2col_array_name = AvailableArrayName( in ProcessTransposeConvOperator()
|
| D | identify_lstm_merge_inputs.cc | 78 AvailableArrayName(*model, base_name + "weights"); in Run()
121 std::string merged_biases = AvailableArrayName(*model, base_name + "biases"); in Run()
165 AvailableArrayName(*model, base_name + "concat_temp"); in Run()
|
| D | reorder_reshape_transpose.cc | 186 AvailableArrayName(*model, transpose_op->outputs[0] + "_exchange"); in Run()
212 AvailableArrayName(*model, transpose_op->inputs[1] + "_copy"); in Run()
230 AvailableArrayName(*model, reshape_op->inputs[1] + "_copy"); in Run()
|
| D | unpartition_embedding_lookup.cc | 182 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_unpartitioned")); in Run()
192 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_permuted/perm")); in Run()
194 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_permuted")); in Run()
|
| D | convert_reorder_axes.cc | 45 AvailableArrayName(*model, reshape_op->outputs[0]); in CreateReshapeFromReorderAxes()
75 AvailableArrayName(*model, transpose_op->outputs[0]); in CreateTransposeFromReorderAxes()
|
| D | identify_nearest_upsample.cc | 194 const std::string shape_array_name = AvailableArrayName(*model, "new_shape"); in Run()
207 AvailableArrayName(*model, "reshape_output"); in Run()
239 AvailableArrayName(*model, absl::StrCat("pack_at_", axis)); in Run()
|
| D | unfuse_activation_functions.cc | 68 AvailableArrayName(*model, op->outputs[0] + "_unfused"); in Run()
|
| D | convert_trivial_pack_to_reshape.cc | 63 AvailableArrayName(*model, pack_op->outputs[0] + "_shape"); in Run()
|
| D | identify_lstm.cc | 278 AvailableArrayName(*model, base_name + "concat_temp"); in Run()
284 AvailableArrayName(*model, base_name + "activ_temp"); in Run()
|
| D | make_initial_dequantize_operator.cc | 67 AvailableArrayName(*model, input_name + "_dequantized"); in AddDequantizeOperatorToInput()
|
| D | ensure_bias_vectors.cc | 66 AvailableArrayName(*model, output_name + "_bias"); in ProcessLinearOperator()
|
| D | convert_expanddims_to_reshape.cc | 81 toco::AvailableArrayName(*model, axis_array_name); in Run()
|
| D | identify_prelu.cc | 111 AvailableArrayName(*model, neg_alpha_tensor_name + "_neg"); in Run()
|
| D | convert_trivial_transpose_to_reshape.cc | 95 toco::AvailableArrayName(*model, perm_array_name); in Run()
|
| D | reorder_elementwise_unary.cc | 119 AvailableArrayName(*model, element_op->outputs[0] + "_reorder"); in Run()
|
| D | lstm_utils.cc | 44 *array_name = AvailableArrayName(*model, *array_name); in CreateFloatArrayBuffer()
|
| D | resolve_reorder_axes.cc | 32 const std::string& newname = AvailableArrayName(*model, desired_newname); in RenameArray()
|
| D | shuffle_fc_weights.cc | 147 AvailableArrayName(*model, fc_op->inputs[0] + "_shuffled"); in Run()
|
| D | dequantize.cc | 155 const std::string& new_array_name = AvailableArrayName(*model, array_name); in DequantizeArray()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | import.cc | 155 toco::AvailableArrayName(*model, "OptionalTensor"); in ImportOperators()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 254 std::string AvailableArrayName(const Model& model, const std::string& name);
|