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()
77 Array& perm_array = model->GetOrCreateArray(perm_array_name); in CreateTransposeFromReorderAxes() local78 *(perm_array.mutable_shape()->mutable_dims()) = { in CreateTransposeFromReorderAxes()80 perm_array.data_type = ArrayDataType::kInt32; in CreateTransposeFromReorderAxes()81 auto& perm_buffer = perm_array.GetMutableBuffer<ArrayDataType::kInt32>(); in CreateTransposeFromReorderAxes()
200 auto& perm_array = in Run() local202 perm_array.data_type = ArrayDataType::kInt32; in Run()203 perm_array.mutable_shape()->ReplaceDims( in Run()205 auto& perm_data = perm_array.GetMutableBuffer<ArrayDataType::kInt32>().data; in Run()206 perm_data.resize(RequiredBufferSizeForShape(perm_array.shape())); in Run()
187 const auto perm_array = GetTransposePerm(input_array); in TransposeInput() local189 model, AvailableArrayName(*model, input + "/transpose/perm"), perm_array); in TransposeInput()195 GetTransposeShape(input_array.shape(), perm_array); in TransposeInput()
1779 auto& perm_array = model->GetArray(op->inputs[1]); in ProcessTransposeOperator() local1780 if (!perm_array.has_shape()) { in ProcessTransposeOperator()1784 if (!perm_array.buffer) { in ProcessTransposeOperator()1788 CHECK(perm_array.data_type == ArrayDataType::kInt32) in ProcessTransposeOperator()1792 perm_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTransposeOperator()
1898 string perm_array = CreateConstArray<ArrayDataType::kInt32>( in ConvertTransposeConvOperator() local1900 transpose->inputs = {weights_name, perm_array}; in ConvertTransposeConvOperator()