Searched refs:perm_array (Results 1 – 6 of 6) sorted by relevance
43 const auto& perm_array = model->GetArray(op->inputs[1]); in Run() local44 if (!perm_array.has_shape()) return ::tensorflow::Status::OK(); in Run()46 const std::vector<int>& perm_dims = perm_array.shape().dims(); in Run()50 perm_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
78 Array& perm_array = model->GetOrCreateArray(perm_array_name); in CreateTransposeFromReorderAxes() local79 *(perm_array.mutable_shape()->mutable_dims()) = { in CreateTransposeFromReorderAxes()81 perm_array.data_type = ArrayDataType::kInt32; in CreateTransposeFromReorderAxes()82 auto& perm_buffer = perm_array.GetMutableBuffer<ArrayDataType::kInt32>(); in CreateTransposeFromReorderAxes()
202 auto& perm_array = in Run() local204 perm_array.data_type = ArrayDataType::kInt32; in Run()205 perm_array.mutable_shape()->ReplaceDims( in Run()207 auto& perm_data = perm_array.GetMutableBuffer<ArrayDataType::kInt32>().data; in Run()208 perm_data.resize(RequiredBufferSizeForShape(perm_array.shape())); in Run()
116 const auto perm_array = GetTransposePerm(input_array); in TransposeInput() local118 model, AvailableArrayName(*model, input + "/transpose/perm"), perm_array); in TransposeInput()124 GetTransposeShape(input_array.shape(), perm_array); in TransposeInput()
1799 auto& perm_array = model->GetArray(op->inputs[1]); in ProcessTransposeOperator() local1800 if (!perm_array.has_shape()) { in ProcessTransposeOperator()1804 if (!perm_array.buffer) { in ProcessTransposeOperator()1808 CHECK(perm_array.data_type == ArrayDataType::kInt32) in ProcessTransposeOperator()1812 perm_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTransposeOperator()
2023 std::string perm_array = CreateConstArray<ArrayDataType::kInt32>( in ConvertTransposeConvOperator() local2025 transpose->inputs = {weights_name, perm_array}; in ConvertTransposeConvOperator()