Searched refs:perm_dims (Results 1 – 2 of 2) sorted by relevance
46 const std::vector<int>& perm_dims = perm_array.shape().dims(); in Run() local47 CHECK_EQ(perm_dims.size(), 1); in Run()51 for (int i = 0; i < perm_dims[0]; ++i) { in Run()
41 std::vector<int64> perm_dims = batch_dims; in LuDecomposition() local42 perm_dims.push_back(m); in LuDecomposition()45 ShapeUtil::MakeShape(S32, perm_dims)}); in LuDecomposition()