Searched refs:AvailableArrayName (Results 1 - 20 of 20) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
H A D | unroll_batch_matmul.cc | 94 slice_a_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_a")}; 106 AvailableArrayName(*model, batch_name + "/slice_a/reshape")}; 121 slice_b_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_b")}; 133 AvailableArrayName(*model, batch_name + "/slice_b/reshape")}; 143 matmul_op->outputs = {AvailableArrayName(*model, batch_name)};
|
H A D | create_im2col_arrays.cc | 57 AvailableArrayName(*model, conv_op->inputs[0] + "_im2col");
|
H A D | ensure_bias_vectors.cc | 34 const string& bias_name = AvailableArrayName(*model, output_name + "_bias");
|
H A D | resolve_batch_normalization.cc | 56 AvailableArrayName(*model, bn_op->outputs[0] + "_mul"); 58 AvailableArrayName(*model, bn_op->outputs[0] + "_add"); 59 const string mul_param_name = AvailableArrayName(*model, mul_name + "_param"); 60 const string add_param_name = AvailableArrayName(*model, add_name + "_param");
|
H A D | unfuse_activation_functions.cc | 65 AvailableArrayName(*model, op->outputs[0] + "_unfused");
|
H A D | resolve_tensorflow_matmul.cc | 44 AvailableArrayName(*model, matmul_op->inputs[1] + "/transpose/perm"), 47 AvailableArrayName(*model, matmul_op->inputs[1] + "/transpose")};
|
H A D | convert_trivial_stack_to_reshape.cc | 60 AvailableArrayName(*model, stack_op->outputs[0] + "_shape");
|
H A D | convert_trivial_transpose_to_reshape.cc | 61 string shape_array_name = toco::AvailableArrayName(*model, perm_array_name);
|
H A D | convert_reorder_axes.cc | 45 AvailableArrayName(*model, reshape_op->outputs[0]); 74 string perm_array_name = AvailableArrayName(*model, transpose_op->outputs[0]);
|
H A D | convert_expanddims_to_reshape.cc | 76 string shape_array_name = toco::AvailableArrayName(*model, axis_array_name);
|
H A D | make_initial_dequantize_operator.cc | 65 AvailableArrayName(*model, input_name + "_dequantized");
|
H A D | lstm_utils.cc | 44 *array_name = AvailableArrayName(*model, *array_name);
|
H A D | identify_lstm_merge_inputs.cc | 73 string merged_weights = AvailableArrayName(*model, base_name + "weights"); 116 string merged_biases = AvailableArrayName(*model, base_name + "biases");
|
H A D | identify_lstm.cc | 334 AvailableArrayName(*model, base_name + "concat_temp"); 338 AvailableArrayName(*model, base_name + "activ_temp");
|
H A D | dequantize.cc | 155 const string& new_array_name = AvailableArrayName(*model, array_name);
|
H A D | quantize.cc | 510 AvailableArrayName(*model, output + "_dequantized");
|
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
H A D | import.cc | 129 toco::AvailableArrayName(*model, "OptionalTensor");
|
/external/tensorflow/tensorflow/contrib/lite/toco/ |
H A D | tooling_util.h | 254 string AvailableArrayName(const Model& model, const string& name);
|
H A D | export_tensorflow.cc | 494 AvailableArrayName(model, src_op.outputs[0] + "/reshape"); 496 AvailableArrayName(model, reshape_output + "/shape"); 520 AvailableArrayName(model, matmul_output + "/transpose_weights"); 522 AvailableArrayName(model, transpose_output + "/perm"); 1869 AvailableArrayName(*model, array_name + "/data");
|
H A D | tooling_util.cc | 1407 string AvailableArrayName(const Model& model, const string& name) { function in namespace:toco 1465 auto param_array_name = AvailableArrayName(*model, param_name);
|
Completed in 317 milliseconds