Home
last modified time | relevance | path

Searched refs:perm_array_name (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_trivial_transpose_to_reshape.cc93 string perm_array_name = transpose_op->inputs[1]; in Run() local
94 string shape_array_name = toco::AvailableArrayName(*model, perm_array_name); in Run()
104 if (IsDiscardableArray(*model, perm_array_name) && in Run()
105 CountOpsWithInput(*model, perm_array_name) == 1) { in Run()
106 model->EraseArray(perm_array_name); in Run()
Dconvert_reorder_axes.cc74 string perm_array_name = AvailableArrayName(*model, transpose_op->outputs[0]); in CreateTransposeFromReorderAxes() local
75 transpose_op->inputs.push_back(perm_array_name); in CreateTransposeFromReorderAxes()
77 Array& perm_array = model->GetOrCreateArray(perm_array_name); in CreateTransposeFromReorderAxes()
Dunroll_batch_matmul.cc188 const string perm_array_name = CreateInt32Array( in TransposeInput() local
191 transpose_op->inputs = {input, perm_array_name}; in TransposeInput()