Home
last modified time | relevance | path

Searched defs:perm_array (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_transpose_attributes.cc43 const auto& perm_array = model->GetArray(op->inputs[1]); in Run() local
H A Dunpartition_embedding_lookup.cc202 auto& perm_array = in Run() local
H A Dconvert_reorder_axes.cc78 Array& perm_array = model->GetOrCreateArray(perm_array_name); in CreateTransposeFromReorderAxes() local
H A Dunroll_batch_matmul.cc118 const auto perm_array = GetTransposePerm(input_array); in TransposeInput() local
H A Dpropagate_fixed_sizes.cc1800 auto& perm_array = model->GetArray(op->inputs[1]); in ProcessTransposeOperator() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dimport_tensorflow.cc2030 std::string perm_array = CreateConstArray<ArrayDataType::kInt32>( in ConvertTransposeConvOperator() local