| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 57 slice_a_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_a")}; in UnrollBatchMatMul3D()
69 AvailableArrayName(*model, batch_name + "/slice_a/reshape")}; in UnrollBatchMatMul3D()
89 slice_b_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_b")}; in UnrollBatchMatMul3D()
101 AvailableArrayName(*model, batch_name + "/slice_b/reshape")}; in UnrollBatchMatMul3D()
111 matmul_op->outputs = {AvailableArrayName(*model, batch_name)}; in UnrollBatchMatMul3D()
151 pack_op->outputs = {AvailableArrayName(*model, batch_name + "/pack")}; in UnrollBatchMatMulRecursion()
189 model, AvailableArrayName(*model, input + "/transpose/perm"), perm_array); in TransposeInput()
192 transpose_op->outputs = {AvailableArrayName(*model, input + "/transpose")}; in TransposeInput()
|
| 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()
69 const string mul_param_name = AvailableArrayName(*model, mul_name + "_param"); in Run()
70 const string add_param_name = 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 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 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()
373 const 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 string& im2col_array_name = AvailableArrayName( in ProcessTransposeConvOperator()
|
| D | identify_lstm_merge_inputs.cc | 77 string merged_weights = AvailableArrayName(*model, base_name + "weights"); in Run()
120 string merged_biases = AvailableArrayName(*model, base_name + "biases"); in Run()
164 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 | 180 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_unpartitioned")); in Run()
190 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_permuted/perm")); in Run()
192 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_permuted")); in Run()
|
| D | convert_reorder_axes.cc | 45 AvailableArrayName(*model, reshape_op->outputs[0]); in CreateReshapeFromReorderAxes()
74 string perm_array_name = AvailableArrayName(*model, transpose_op->outputs[0]); in CreateTransposeFromReorderAxes()
|
| D | resolve_tensorflow_matmul.cc | 103 model, AvailableArrayName(*model, input_rhs + "/transpose/perm"), in Run()
106 AvailableArrayName(*model, input_rhs + "/transpose")}; in Run()
|
| D | unfuse_activation_functions.cc | 68 AvailableArrayName(*model, op->outputs[0] + "_unfused"); 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 | convert_trivial_pack_to_reshape.cc | 63 AvailableArrayName(*model, pack_op->outputs[0] + "_shape"); in Run()
|
| D | ensure_bias_vectors.cc | 53 const string& bias_name = AvailableArrayName(*model, output_name + "_bias"); in ProcessLinearOperator()
|
| D | make_initial_dequantize_operator.cc | 66 AvailableArrayName(*model, input_name + "_dequantized"); in AddDequantizeOperatorToInput()
|
| D | convert_expanddims_to_reshape.cc | 74 string shape_array_name = toco::AvailableArrayName(*model, axis_array_name); in Run()
|
| D | convert_trivial_transpose_to_reshape.cc | 94 string shape_array_name = toco::AvailableArrayName(*model, perm_array_name); in Run()
|
| D | identify_prelu.cc | 111 AvailableArrayName(*model, neg_alpha_tensor_name + "_neg"); in Run()
|
| D | lstm_utils.cc | 44 *array_name = AvailableArrayName(*model, *array_name); in CreateFloatArrayBuffer()
|
| D | resolve_reorder_axes.cc | 32 const string& newname = AvailableArrayName(*model, desired_newname); in RenameArray()
|
| D | reorder_elementwise_unary.cc | 118 AvailableArrayName(*model, element_op->outputs[0] + "_reorder"); in Run()
|
| D | shuffle_fc_weights.cc | 147 AvailableArrayName(*model, fc_op->inputs[0] + "_shuffled"); in Run()
|
| D | dequantize.cc | 155 const string& new_array_name = AvailableArrayName(*model, array_name); in DequantizeArray()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | import.cc | 154 toco::AvailableArrayName(*model, "OptionalTensor"); in ImportOperators()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 256 string AvailableArrayName(const Model& model, const string& name);
|
| D | tooling_util.cc | 1264 AvailableArrayName(*model, output_array_name + "_copy"); in FixEdgeArrays()
1367 model, AvailableArrayName(*model, target_array_name + "_copy_shape"), in InsertCopyOperator()
1802 string AvailableArrayName(const Model& model, const string& name) { in AvailableArrayName() function
1860 auto param_array_name = AvailableArrayName(*model, param_name); in CreateInt32Array()
|